Fix line break style

cooper-org · May 1, 2024 · 0b32859 · 0b32859
1 parent ac98a07
commit 0b32859
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/helpers/cooper_test_utils.py b/tests/helpers/cooper_test_utils.py
@@ -104,9 +104,9 @@ def _compute_eq_or_ineq_violations(self, x, A_or_C, b_or_d, has_constraint, use_
         strict_violation = None
         strict_constraint_features = None
         if use_surrogate:
-            strict_violation = (
-                A_or_C + 0.1 * torch.randn(A_or_C.shape, generator=self.generator, device=self.device)
-            ) @ x
+            strict_violation = torch.mm(
+                A_or_C + 0.1 * torch.randn(A_or_C.shape, generator=self.generator, device=self.device), x
+            )
             if multiplier_type == cooper.multipliers.IndexedMultiplier:
                 strict_constraint_features = torch.randperm(
                     b_or_d.numel(), generator=self.generator, device=self.device