Fixed bug in sort

lrcfmd · Mar 6, 2021 · 18209e3 · 18209e3
1 parent 387985f
commit 18209e3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ElM2D/ElM2D.py b/ElM2D/ElM2D.py
@@ -251,7 +251,7 @@ def PCA(self, n_components=5):
 
         return Y[:, :n_components]
 
-    def sort_compositions(self, X):
+    def sort_compositions(self, X=None):
         """
         Parameters: X, an iterable list of compositions in string format
         Returns: The indices of the input list as they fall in sorted order compositionally
@@ -261,7 +261,7 @@ def sort_compositions(self, X):
         sorted_indices = ChemMapper.sort_compositions(comps)
         sorted_comps = comps[sorted_indices]
         """
-        if self.dm == None:
+        if self.dm is None:
             self.fit(X)
 
         dists_1D = self.PCA(n_components=1)