From 680848b9285d3bbd57177db974e615ad1733860e Mon Sep 17 00:00:00 2001 From: kianorr Date: Thu, 4 Jul 2024 16:26:16 -0400 Subject: [PATCH] add QuadraticFlux obj - doesn't pass due to jax bug, maybe PR #1002 will fix it? --- tests/test_examples.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tests/test_examples.py b/tests/test_examples.py index bf55c6058a..4639b72e62 100644 --- a/tests/test_examples.py +++ b/tests/test_examples.py @@ -1351,7 +1351,10 @@ def test_optimize_with_all_coil_types(DummyCoilSet, DummyMixedCoilSet): def test(c, method): target = 11 - obj = ObjectiveFunction(CoilLength(c, target=target)) + obj = ObjectiveFunction( + CoilLength(c, target=target), + QuadraticFlux(eq=Equilibrium(), field=c, vacuum=True), + ) optimizer = Optimizer(method) (c,), _ = optimizer.optimize(c, obj, maxiter=200, ftol=0, xtol=1e-15) flattened_coils = tree_leaves(