Support flow volumes with negative delta_z in reconcile_flows.

PiperOrigin-RevId: 580548736
google-research · Nov 8, 2023 · b9cfb56 · b9cfb56
1 parent 5a1eb15
commit b9cfb56
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flow_utils.py b/flow_utils.py
@@ -104,7 +104,7 @@ def reconcile_flows(flows: Sequence[np.ndarray], max_gradient: float,
     # Try to fill any invalid values.
     m = np.repeat(np.isnan(ret[0:1, ...]), ret.shape[0], 0)
     if ret.shape[0] == 3:
-      m &= np.repeat(f[2:3, ...] >= min_delta_z, 3, 0)
+      m &= np.repeat(np.abs(f[2:3, ...]) >= min_delta_z, 3, 0)
     ret[m] = f[m]
 
   if max_gradient > 0: