Fixed elastica tests.

alucantonio · Mar 5, 2024 · f614076 · f614076
1 parent 1869a87
commit f614076
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_elastica.py b/tests/test_elastica.py
@@ -242,7 +242,7 @@ def energy(theta: npt.NDArray, B, theta_0, F) -> Array:
 
         residual = C.sub(C.codifferential(C.star(moment)), load)
 
-        return residual.coeffs[1:]
+        return residual.coeffs.flatten()[1:]
 
     energy_grad = energy