sympy: proper version check

devitocodes · Oct 5, 2021 · c598d78 · c598d78
1 parent 624e3ba
commit c598d78
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/devito/finite_differences/elementary.py b/devito/finite_differences/elementary.py
@@ -1,5 +1,7 @@
 import sympy
 
+from distutils import version
+
 from devito.finite_differences.differentiable import DifferentiableFunction, diffify
 
 
@@ -646,7 +648,7 @@ class mathieucprime(DifferentiableFunction, sympy.mathieucprime):
 
 
 # New elementary functions in sympy 1.8
-if float(sympy.__version__) >= 1.8:
+if version.LooseVersion(sympy.__version__) >= version.LooseVersion('1.8'):
 
     class motzkin(DifferentiableFunction, sympy.motzkin):
         __sympy_class__ = sympy.motzkin