From aaf4cdd513b38fcb4d0d24f737f4aefc169fc410 Mon Sep 17 00:00:00 2001
From: Thibaut Lunet <thibaut.lunet@tuhh.de>
Date: Sat, 18 Jan 2025 22:30:11 +0100
Subject: [PATCH] TL: allowed ndarray.dtype for FieldsIO arg
---
pySDC/helpers/fieldsIO.py | 9 +++++++--
1 file changed, 7 insertions(+), 2 deletions(-)
diff --git a/pySDC/helpers/fieldsIO.py b/pySDC/helpers/fieldsIO.py
index 4507a8bdb..bec1e73de 100644
--- 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