Added check for scikits.sparse

jellis18 · Jan 18, 2017 · dfc37b2 · dfc37b2
1 parent ccf9a92
commit dfc37b2
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/PAL2/PALmodels.py b/PAL2/PALmodels.py
@@ -32,9 +32,16 @@
     from sksparse.cholmod import cholesky
     SK_SPARSE = True
 except ImportError:
-    print 'WARNING: scikit-sparse note installed, will not use sparse matrices'
-    SK_SPARSE=False
-
+    print 'WARNING: scikit-sparse note installed, will try scikits.sparse...'
+    SK_SPARSE = False
+
+if not SK_SPARSE:
+    try:
+        from scikits.sparse.cholmod import cholesky
+        SK_SPARSE = True
+    except ImportError:
+        print 'WARNING: scikits.sparse note installed, will not use sparse matrices'
+        SK_SPARSE = False
 
 import matplotlib.pyplot as plt