From cdd88830f834eb8fe8145e216bbb8a150ffdb20b Mon Sep 17 00:00:00 2001
From: Felix Koehler <f.koehler@tum.de>
Date: Tue, 27 Feb 2024 13:01:14 +0100
Subject: [PATCH] Fix wrong imports

---
 exponax/sample_stepper/burgers.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/exponax/sample_stepper/burgers.py b/exponax/sample_stepper/burgers.py
index f800734..6dec779 100644
--- a/exponax/sample_stepper/burgers.py
+++ b/exponax/sample_stepper/burgers.py
@@ -1,4 +1,4 @@
-from jax import Array, Complex
+from jaxtyping import Array, Complex
 
 from ..base_stepper import BaseStepper
 from ..nonlinear_functions import ConvectionNonlinearFun