fix Riemann TGV

tumaer · Jun 12, 2024 · a7f3108 · a7f3108
1 parent 45409b9
commit a7f3108
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 9 deletions.
diff --git a/docs/requirements.txt b/docs/requirements.txt
@@ -1,13 +1,13 @@
+absl-py>=2.1.0
 h5py
 jax[cpu]==0.4.29
+
+# JAX-MD dependencies
+jraph>=0.0.6.dev0
 omegaconf
 pandas
 pyvista
 sphinx==7.2.6
 sphinx-exec-code==0.12
 sphinx-rtd-theme==1.3.0
 toml==0.10.2
-
-# JAX-MD dependencies
-jraph>=0.0.6.dev0
-absl-py>=2.1.0
diff --git a/jax_sph/case_setup.py b/jax_sph/case_setup.py
@@ -154,7 +154,9 @@ def initialize(self):
             num_particles, mass_ref, cfg.case
         )
         # whether to compute wall normals
-        is_nw = cfg.solver.free_slip or cfg.solver.name == "RIE"
+        is_nw = (
+            cfg.solver.free_slip or cfg.solver.name == "RIE"
+        ) and cfg.solver.is_bc_trick
         # calculate wall normals if necessary
         nw = self._compute_wall_normals("scipy")(r, tag) if is_nw else jnp.zeros_like(r)
 

diff --git a/requirements.txt b/requirements.txt
@@ -1,9 +1,9 @@
+absl-py>=2.1.0
 h5py
 jax[cpu]==0.4.29
-omegaconf
-pandas
-pyvista
 
 # JAX-MD dependencies
 jraph>=0.0.6.dev0
-absl-py>=2.1.0
+omegaconf
+pandas
+pyvista