Fixing error related nump array in store_transport_to_hdf

tardis-sn · Jun 14, 2024 · 54d99f8 · 54d99f8
1 parent 22dd694
commit 54d99f8
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/tardis/io/model_reader.py b/tardis/io/model_reader.py
@@ -637,8 +637,13 @@ def store_transport_to_hdf(transport, fname):
 
         for key, value in transport_data.items():
             if key.endswith("_cgs"):
-                transport_group.create_dataset(key, data=value[0])
-                transport_group.create_dataset(key + "_unit", data=value[1])
+                if isinstance(value, tuple) and len(value) == 2:
+                    transport_group.create_dataset(key, data=value[0])
+                    transport_group.create_dataset(key + "_unit", data=value[1])
+                else:
+                    transport_group.create_dataset(key, data=value)
+                # transport_group.create_dataset(key, data=value[0])
+                # transport_group.create_dataset(key + "_unit", data=value[1])
             else:
                 if value is not None:
                     transport_group.create_dataset(key, data=value)