Can't use copy=None on some older NumPy versions...

rapidsai · Jul 16, 2024 · 72f456f · 72f456f
1 parent 1e912c0
commit 72f456f
Showing 1 changed file with 10 additions and 9 deletions.
diff --git a/python/cuml/internals/array.py b/python/cuml/internals/array.py
@@ -1168,19 +1168,20 @@ def from_input(
             )
 
         make_copy = force_contiguous and not arr.is_contiguous
-        if not make_copy:
-            # NumPy now interprets False as never copy, so must use None
-            make_copy = None
 
         if (
             not fail_on_order and order != arr.order and order != "K"
         ) or make_copy:
-            arr = cls(
-                arr.mem_type.xpy.array(
-                    arr.to_output("array"), order=order, copy=make_copy
-                ),
-                index=index,
-            )
+            if make_copy:
+                data = arr.mem_type.xpy.array(
+                    arr.to_output("array"), order=order
+                )
+            else:
+                data = arr.mem_type.xpy.asarray(
+                    arr.to_output("array"), order=order
+                )
+
+            arr = cls(data, index=index)
 
         n_rows = arr.shape[0]