minor fixes

Jammy2211 · Jammy2211 · commit c249e3b3b407 · 2025-10-16T15:42:09.000+01:00
diff --git a/autoarray/operators/transformer.py b/autoarray/operators/transformer.py
@@ -1,4 +1,5 @@
 import copy
+import jax
 import jax.numpy as jnp
 import numpy as np
 import warnings
