checking to see if dataset_name is in keys before attempting to remov…

…e it #319
ORNL · Feb 5, 2025 · 4c3e53c · 4c3e53c
1 parent c62961d
commit 4c3e53c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/hydragnn/utils/datasets/adiosdataset.py b/hydragnn/utils/datasets/adiosdataset.py
@@ -174,7 +174,10 @@ def save(self):
             if len(self.dataset[label]) > 0:
                 data = self.dataset[label][0]
                 keys = data.keys() if callable(data.keys) else data.keys
-                keys.remove("dataset_name")  # we dont need this to be added to the keys
+                if "dataset_name" in keys:
+                    keys.remove(
+                        "dataset_name"
+                    )  # we dont need this to be added to the keys
                 self.io.DefineAttribute("%s/keys" % label, keys)
                 keys = sorted(keys)
                 self.comm.allgather(keys)