check magnitude is array

hgrecco · Dec 24, 2023 · d331aaf · d331aaf
1 parent f9e139e
commit d331aaf
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/pint/facets/numpy/quantity.py b/pint/facets/numpy/quantity.py
@@ -115,11 +115,12 @@ def _numpy_method_wrap(self, func, *args, **kwargs):
         return value
 
     def __array__(self, t=None) -> np.ndarray:
-        warnings.warn(
-            "The unit of the quantity is stripped when downcasting to ndarray.",
-            UnitStrippedWarning,
-            stacklevel=2,
-        )
+        if isinstance(self._magnitude, np.ndarray):
+            warnings.warn(
+                "The unit of the quantity is stripped when downcasting to ndarray.",
+                UnitStrippedWarning,
+                stacklevel=2,
+            )
         return _to_magnitude(self._magnitude, force_ndarray=True)
 
     def clip(self, min=None, max=None, out=None, **kwargs):