From 7dc223a364dac7b5608b7969fe9111f49ef608cb Mon Sep 17 00:00:00 2001 From: Jeremy Magland Date: Thu, 4 Apr 2024 14:04:38 -0400 Subject: [PATCH] float type checking --- lindi/conversion/attr_conversion.py | 2 +- lindi/conversion/nan_inf_ninf.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/lindi/conversion/attr_conversion.py b/lindi/conversion/attr_conversion.py index 15416fa..b6335b3 100644 --- a/lindi/conversion/attr_conversion.py +++ b/lindi/conversion/attr_conversion.py @@ -23,7 +23,7 @@ def h5_to_zarr_attr(attr: Any, *, label: str = '', h5f: Union[h5py.File, None]): raise Exception(f"Unexpected h5 attribute: None at {label}") elif type(attr) in [int, np.int8, np.int16, np.int32, np.int64, np.uint8, np.uint16, np.uint32, np.uint64]: return int(attr) - elif type(attr) in [float, np.float16, np.float32, np.float64]: + elif isinstance(attr, (float, np.floating)): return encode_nan_inf_ninf(float(attr)) elif isinstance(attr, complex) or (isinstance(attr, np.ndarray) and np.issubdtype(attr.dtype, np.complexfloating)): raise Exception(f"Complex number is not supported at {label}") diff --git a/lindi/conversion/nan_inf_ninf.py b/lindi/conversion/nan_inf_ninf.py index 3ab012b..cc7595d 100644 --- a/lindi/conversion/nan_inf_ninf.py +++ b/lindi/conversion/nan_inf_ninf.py @@ -21,7 +21,7 @@ def encode_nan_inf_ninf(val): return [encode_nan_inf_ninf(v) for v in val] elif isinstance(val, dict): return {k: encode_nan_inf_ninf(v) for k, v in val.items()} - elif type(val) in [float, np.float16, np.float32, np.float64]: + elif isinstance(val, (float, np.floating)): if np.isnan(val): return 'NaN' elif val == float('inf'):