fixup torch

soda-inria · Jan 18, 2024 · d2ccf5b · d2ccf5b
1 parent d5bab3f
commit d2ccf5b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/benchmarks/ridge/solvers/sklearn_torch_dispatch.py b/benchmarks/ridge/solvers/sklearn_torch_dispatch.py
@@ -87,7 +87,7 @@ def warm_up(self):
         n_warmup_features = 5
         sample_weight = self.sample_weight
         if sample_weight is not None:
-            sample_weight = sample_weight[:n_warmup_samples].copy()
+            sample_weight = sample_weight[:n_warmup_samples].clone()
         with config_context(array_api_dispatch=True):
             Ridge(
                 alpha=self.alpha,
@@ -99,8 +99,8 @@ def warm_up(self):
                 positive=True if (self.solver == "lbfgs") else False,
                 random_state=self.random_state,
             ).fit(
-                self.X[:n_warmup_samples, :n_warmup_features].copy(),
-                self.y[:n_warmup_samples].copy(),
+                self.X[:n_warmup_samples, :n_warmup_features].clone(),
+                self.y[:n_warmup_samples].clone(),
                 sample_weight,
             )