Skip to content

Commit aaf4cdd

Browse files
committed
TL: allowed ndarray.dtype for FieldsIO arg
1 parent 4f94845 commit aaf4cdd

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

pySDC/helpers/fieldsIO.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -100,8 +100,13 @@ def __init__(self, dtype, fileName):
100100
fileName : str
101101
File.
102102
"""
103-
assert dtype in DTYPES_AVAIL, f"{dtype=} not available"
104-
self.dtype = dtype
103+
avail = False
104+
for key in DTYPES_AVAIL:
105+
if key == dtype:
106+
self.dtype = key
107+
avail = True
108+
break
109+
assert avail, f"{dtype=} not available"
105110
self.fileName = fileName
106111
self.initialized = False
107112

0 commit comments

Comments
 (0)