another fix

XanaduAI · Dec 20, 2024 · ea9ec22 · ea9ec22
1 parent 6917c62
commit ea9ec22
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_training/test_opt_lab_dev.py b/tests/test_training/test_opt_lab_dev.py
@@ -461,7 +461,7 @@ def test_opt_backend_param(self):
 
         def cost_fn_sympl():
             state_out = Vacuum((0,)) >> S >> Rgate((0,), theta=r_angle)
-            return 1 - math.pow(state_out >> target_state.dual, 2)
+            return 1 - math.abs((state_out >> target_state.dual) ** 2)
 
         opt = Optimizer(symplectic_lr=0.1, euclidean_lr=0.05)
         opt.minimize(cost_fn_sympl, by_optimizing=[S, r_angle])