diff --git a/autoarray/numpy_wrapper.py b/autoarray/numpy_wrapper.py index 60758e97..81564bdb 100644 --- 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"