importfix

Author: dbezgin

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