From 0b328598f5442ad49c458bcb2f061f6b90edb0c8 Mon Sep 17 00:00:00 2001 From: Meraj Hashemizadeh Date: Wed, 1 May 2024 16:54:48 -0400 Subject: [PATCH] Fix line break style --- tests/helpers/cooper_test_utils.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/helpers/cooper_test_utils.py b/tests/helpers/cooper_test_utils.py index 4fc313f4..46bd5509 100644 --- 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