single line fix

lhatsk · Dec 14, 2023 · ace62d6 · ace62d6
1 parent 5e93421
commit ace62d6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/contacts_to_distograms.py b/contacts_to_distograms.py
@@ -35,7 +35,7 @@ def main():
 
     contacts = np.loadtxt(args.csv)
 
-    if len(contacts.shape) < 2:
+    if contacts.size <= 1:
         contacts = np.array([contacts])
 
     with open(args.output, 'w') as f:

diff --git a/preprocessing_distributions.py b/preprocessing_distributions.py
@@ -43,7 +43,7 @@
                            dtype=None,
                            encoding=None)
 
-if len(restraints.shape) < 2:
+if restraints.size <= 1:
     restraints = np.array([restraints])
 
 distogram = []