added import handling for joint py2/3 support

tclose · Oct 23, 2017 · 1326122 · 1326122
1 parent f7e0178
commit 1326122
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/diophantine.py b/diophantine.py
@@ -40,7 +40,11 @@
 from builtins import next
 from builtins import range
 from copy import deepcopy
-from math import copysign, sqrt, log10, floor, gcd
+from math import copysign, sqrt, log10, floor
+try:
+    from math import gcd  # Py >= 3.6
+except ImportError:
+    from fractions import gcd  # Py < 3.6
 from sympy import Matrix, zeros, ones, eye
 from itertools import chain, product