Get avg flows for relevant subproblem only

blue-marble · Jan 24, 2025 · 1422b5d · 1422b5d
1 parent 0603f61
commit 1422b5d
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/gridpath/system/water/water_flows.py b/gridpath/system/water/water_flows.py
@@ -510,9 +510,10 @@ def get_inputs_from_database(
                 {subscenarios.WATER_NETWORK_SCENARIO_ID}
             )
             AND (balancing_type, horizon)
-            IN (SELECT balancing_type, horizon
-                FROM inputs_temporal_horizons
+            IN (SELECT DISTINCT balancing_type, horizon
+                FROM inputs_temporal_horizon_timepoints
                 WHERE temporal_scenario_id = {subscenarios.TEMPORAL_SCENARIO_ID}
+                AND subproblem_id = {subproblem}
                 )
             ;
             """
@@ -536,9 +537,10 @@ def get_inputs_from_database(
                 {subscenarios.WATER_NETWORK_SCENARIO_ID}
             )
             AND (balancing_type, horizon)
-            IN (SELECT balancing_type, horizon
-                FROM inputs_temporal_horizons
+            IN (SELECT DISTINCT balancing_type, horizon
+                FROM inputs_temporal_horizon_timepoints
                 WHERE temporal_scenario_id = {subscenarios.TEMPORAL_SCENARIO_ID}
+                AND subproblem_id = {subproblem}
                 )
             ;
             """