diff --git a/benchmarks/ridge/solvers/sklearn_torch_dispatch.py b/benchmarks/ridge/solvers/sklearn_torch_dispatch.py index f51f829..1ba6213 100644 --- 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, )