Traceback (most recent call last):
  File "<stdin>", line 118, in <module>
  File "<stdin>", line 89, in energy
  File "/env/lib/python3.12/site-packages/numpy/_core/einsumfunc.py", line 1423, in einsum
    return c_einsum(*operands, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
ValueError: output has more dimensions than subscripts given in einstein sum, but no '...' ellipsis provided to broadcast the extra dimensions.
