TL: allowed ndarray.dtype for FieldsIO arg

Parallel-in-Time · Jan 18, 2025 · aaf4cdd · aaf4cdd
1 parent 4f94845
commit aaf4cdd
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/pySDC/helpers/fieldsIO.py b/pySDC/helpers/fieldsIO.py
@@ -100,8 +100,13 @@ def __init__(self, dtype, fileName):
         fileName : str
             File.
         """
-        assert dtype in DTYPES_AVAIL, f"{dtype=} not available"
-        self.dtype = dtype
+        avail = False
+        for key in DTYPES_AVAIL:
+            if key == dtype:
+                self.dtype = key
+                avail = True
+                break
+        assert avail, f"{dtype=} not available"
         self.fileName = fileName
         self.initialized = False