importfix

tumaer · Dec 22, 2023 · b5396c2 · b5396c2
1 parent c820856
commit b5396c2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/jaxfluids/solvers/riemann_solvers/LaxFriedrichs.py b/src/jaxfluids/solvers/riemann_solvers/LaxFriedrichs.py
@@ -29,6 +29,8 @@
 #*                                                                              *
 #*------------------------------------------------------------------------------*
 
+import jax.numpy as jnp
+
 from jaxfluids.solvers.riemann_solvers.riemann_solver import RiemannSolver
 from jaxfluids.materials.material_manager import MaterialManager