Allow Trajectory correctly handle system generated from file (#45)

OpenBioSim · Apr 30, 2024 · 61feddc · 61feddc
1 parent b074d06
commit 61feddc
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/python/BioSimSpace/Sandpit/Exscientia/Trajectory/_trajectory.py b/python/BioSimSpace/Sandpit/Exscientia/Trajectory/_trajectory.py
@@ -1109,6 +1109,10 @@ def _split_molecules(frame, pdb, reference, work_dir, property_map={}):
     # Store the formats associated with the reference system.
     formats = reference.fileFormat()
 
+    # Convert NoneType to empty list.
+    if formats is None:
+        formats = []
+
     # Write the frame coordinates/velocities to file.
     coord_file = work_dir + f"/{str(_uuid.uuid4())}.coords"
     top_file = work_dir + f"/{str(_uuid.uuid4())}.top"