Merge pull request #155 from Jammy2211/feature/wrapper_fix

fix issue causing ndarray to not be a type when accessed via jax wrapper
Jammy2211 · Dec 16, 2024 · e7a66fe · e7a66fe
2 parents 28de257 + bae9503
commit e7a66fe
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/autoarray/numpy_wrapper.py b/autoarray/numpy_wrapper.py
@@ -51,6 +51,10 @@ def __getattr__(self, item):
             return Callable(attribute)
         return attribute
 
+    @property
+    def ndarray(self):
+        return np.ndarray
+
 
 use_jax = environ.get("USE_JAX", "0") == "1"