Backport PR #28441: MNT: Update basic units example to work with numpy 2.0

rcomer · meeseeksmachine · commit e46044db0fc6 · 2024-06-24T09:05:01.000Z
diff --git a/galleries/examples/units/basic_units.py b/galleries/examples/units/basic_units.py
@@ -146,10 +146,10 @@ def __getattribute__(self, name):
             return getattr(variable, name)
         return object.__getattribute__(self, name)
 
-    def __array__(self, dtype=object):
+    def __array__(self, dtype=object, copy=False):
         return np.asarray(self.value, dtype)
 
-    def __array_wrap__(self, array, context):
+    def __array_wrap__(self, array, context=None, return_scalar=False):
         return TaggedValue(array, self.unit)
 
     def __repr__(self):
@@ -222,10 +222,10 @@ def __mul__(self, rhs):
     def __rmul__(self, lhs):
         return self*lhs
 
-    def __array_wrap__(self, array, context):
+    def __array_wrap__(self, array, context=None, return_scalar=False):
         return TaggedValue(array, self)
 
-    def __array__(self, t=None, context=None):
+    def __array__(self, t=None, context=None, copy=False):
         ret = np.array(1)
         if t is not None:
             return ret.astype(t)
