From 7fd12087d0e1383e0a575c2584bf2f32bb726731 Mon Sep 17 00:00:00 2001 From: Kevin Michael Dalton Date: Wed, 18 Sep 2024 07:52:33 -0700 Subject: [PATCH] allow nan inference for float64 --- reciprocalspaceship/dtypes/internals.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/reciprocalspaceship/dtypes/internals.py b/reciprocalspaceship/dtypes/internals.py index 8d6c7205..c9d78d50 100644 --- a/reciprocalspaceship/dtypes/internals.py +++ b/reciprocalspaceship/dtypes/internals.py @@ -1359,7 +1359,7 @@ def round(self: T, decimals: int = 0, *args, **kwargs) -> T: @wraps(libmissing.is_numeric_na) def is_numeric_na(values): - allowed_dtypes = ("float32", "int32") + allowed_dtypes = ("float64", "float32", "int32") if isinstance(values, np.ndarray) and values.dtype in allowed_dtypes: return np.isnan(values) return libmissing.is_numeric_na(values)