Converts numpy numeric arguments to allow serialization (#86)

RedesignScience · Apr 5, 2024 · 2acd3ff · 2acd3ff
1 parent 6df7254
commit 2acd3ff
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/cvpack/utils.py b/cvpack/utils.py
@@ -220,6 +220,10 @@ def preprocess_args(func: t.Callable) -> t.Callable:
     signature = inspect.signature(func)
 
     def convert(data: t.Any) -> t.Any:  # pylint: disable=too-many-return-statements
+        if isinstance(data, np.integer):
+            return int(data)
+        if isinstance(data, np.floating):
+            return float(data)
         if isinstance(data, mmunit.Quantity):
             return Quantity(data)
         if isinstance(data, mmunit.Unit):