Added support for older versions of gemmi

rigdenlab · Jan 16, 2025 · 8537588 · 8537588
1 parent e791d8c
commit 8537588
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/simbad/util/pdb_util.py b/simbad/util/pdb_util.py
@@ -131,7 +131,10 @@ def keep_first_chain_only(self):
         self.select_chain_by_idx(0)
 
     def keep_first_model_only(self):
-        models = [m.num for m in self.structure]
+        if int(gemmi.__version__.replace(".","")) <= 70:
+            models = [m.name for m in self.structure]
+        else:
+            models = [m.num for m in self.structure]
         for model in models[1:]:
             del self.structure[model]