Correct check of dists in writePickle

bacpop · Aug 7, 2024 · 04c5283 · 04c5283
1 parent 9fb29df
commit 04c5283
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/PopPUNK/utils.py b/PopPUNK/utils.py
@@ -153,7 +153,7 @@ def storePickle(rlist, qlist, self, X, pklName):
     """
     with open(pklName + ".pkl", 'wb') as pickle_file:
         pickle.dump([rlist, qlist, self], pickle_file)
-    if X != None:
+    if isinstance(X, np.ndarray):
         np.save(pklName + ".npy", X)