fix loc

hgrecco · Jan 6, 2025 · 2da273d · 2da273d
1 parent 8c2fc78
commit 2da273d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 7 deletions.
diff --git a/bors.toml b/bors.toml
diff --git a/pint_pandas/pint_array.py b/pint_pandas/pint_array.py
@@ -398,6 +398,8 @@ def _convert_np_result(self, result):
             return result
         elif pd.api.types.is_bool_dtype(result):
             return result
+        elif isinstance(result, numbers.Number):
+            return result
         else:
             # one return value
             return type(self)(result)