[numpy] Fix test failures under NumPy 2.1.

PiperOrigin-RevId: 680790659 Change-Id: I65134d80df59c54b2601bb2445f43f3d75dcac5a
google · Oct 1, 2024 · d465a9c · d465a9c
1 parent 1917a84
commit d465a9c
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/python/tensorstore/tensorstore_class.cc b/python/tensorstore/tensorstore_class.cc
@@ -871,7 +871,7 @@ operation.
 
   cls.def(
       "__array__",
-      [](Self& self, std::optional<py::dtype> dtype,
+      [](Self& self, std::optional<py::dtype> dtype, std::optional<bool> copy,
          std::optional<py::object> context) {
         return ValueOrThrow(internal_python::InterruptibleWait(
             tensorstore::Read<zero_origin>(self.value)));
@@ -920,7 +920,8 @@ See also:
   I/O
 
 )",
-      py::arg("dtype") = std::nullopt, py::arg("context") = std::nullopt);
+      py::arg("dtype") = std::nullopt, py::arg("copy") = std::nullopt,
+      py::arg("context") = std::nullopt);
 
   cls.def(
       "resolve",