Fix version check

pybop-team · Mar 21, 2024 · e513c33 · e513c33
1 parent 76b5c19
commit e513c33
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/unit/test_plots.py b/tests/unit/test_plots.py
@@ -1,6 +1,7 @@
 import pybop
 import numpy as np
 import pytest
+from packaging import version
 
 
 class TestPlots:
@@ -125,7 +126,7 @@ def test_optim_plots(self, optim):
     def test_with_ipykernel(self, dataset, cost, optim):
         import ipykernel
 
-        assert ipykernel.__version__ >= 0.6
+        assert version.parse(ipykernel.__version__) >= version.parse("0.6")
         pybop.plot_dataset(dataset, signal=["Voltage [V]"])
         pybop.plot2d(cost, gradient=True, steps=5)
         pybop.plot_convergence(optim)