From 31a679935c76e4b4c761d463510266b8d48dec08 Mon Sep 17 00:00:00 2001 From: Ianna Osborne Date: Mon, 19 Aug 2024 13:05:40 +0200 Subject: [PATCH] fix: numpy version --- src/ragged/_helper_functions.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/ragged/_helper_functions.py b/src/ragged/_helper_functions.py index 451aa3c..7688543 100644 --- a/src/ragged/_helper_functions.py +++ b/src/ragged/_helper_functions.py @@ -6,7 +6,7 @@ def regularise_to_float(t: np.dtype, /) -> np.dtype: # Ensure compatibility with numpy 2.0.0 - if np.__version__ != "2.0.0": + if np.__version__ >= "2.1": # Just pass and return the input type if the numpy version is not 2.0.0 return t