From 5e934211c7e7efac4f44a8c3ed3985e204ff2956 Mon Sep 17 00:00:00 2001
From: kstahl <kolja.stahl@tu-berlin.de>
Date: Thu, 14 Dec 2023 11:27:33 +0100
Subject: [PATCH] fix for single line restraints

---
 contacts_to_distograms.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/contacts_to_distograms.py b/contacts_to_distograms.py
index 30944ed..8dc2eae 100644
--- 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) == 1:
+    if len(contacts.shape) < 2:
         contacts = np.array([contacts])
 
     with open(args.output, 'w') as f: