Add split to dtype string for unwanted prefix

liamchalcroft · Apr 6, 2023 · 7dd6146 · 7dd6146
1 parent 77c41a8
commit 7dd6146
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/monai/utils/type_conversion.py b/monai/utils/type_conversion.py
@@ -45,7 +45,7 @@
 
 def get_numpy_dtype_from_string(dtype: str) -> np.dtype:
     """Get a numpy dtype (e.g., `np.float32`) from its string (e.g., `"float32"`)."""
-    return np.empty([], dtype=dtype).dtype
+    return np.empty([], dtype=str(dtype).split('.')[-1]).dtype
 
 
 def get_torch_dtype_from_string(dtype: str) -> torch.dtype: