fix for single line restraints

lhatsk · Dec 14, 2023 · 455c379 · 455c379
1 parent 59dc08b
commit 455c379
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/preprocessing_distributions.py b/preprocessing_distributions.py
@@ -43,7 +43,7 @@
                            dtype=None,
                            encoding=None)
 
-if len(restraints.shape) == 1:
+if len(restraints.shape) < 2:
     restraints = np.array([restraints])
 
 distogram = []