Fix unit test.

facebookresearch · Dec 5, 2022 · 83b19b0 · 83b19b0
1 parent bfa60c7
commit 83b19b0
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/optimizer/nonlinear/common.py b/tests/optimizer/nonlinear/common.py
@@ -237,6 +237,9 @@ def _atb_impl(self) -> torch.Tensor:
         def Av(self, v):
             pass
 
+        def diagonal_scaling(self, v):
+            pass
+
     class MockCostFunction(th.CostFunction):
         def __init__(self, optim_vars, cost_weight):
             super().__init__(cost_weight)