do not change input flows in tdp when update_pflow=False

Tractables · Aug 22, 2024 · fb0eaa3 · fb0eaa3
1 parent ec59bd4
commit fb0eaa3
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/pyjuice/model/backend/top_down_prob.py b/src/pyjuice/model/backend/top_down_prob.py
@@ -333,7 +333,8 @@ def eval_top_down_probs(pc, update_pflow: bool = True, scale: float = 1.0, pc_is
                                        pc_is_normalized = pc_is_normalized)
 
     # Backward pass over the input layers
-    for layer in pc.input_layer_group:
-        layer.add_missing_flows(node_flows, scale = scale)
+    if update_pflow:
+        for layer in pc.input_layer_group:
+            layer.add_missing_flows(node_flows, scale = scale)
 
     return None