Hotfix #2 specify is_long where needed

Author: staadecker

switch_model/generators/core/dispatch.py

@@ -581,7 +581,7 @@ def c(func):
     )
 
 
-@graph("dispatch", title="Average daily dispatch")
+@graph("dispatch", title="Average daily dispatch", is_long=True)
 def graph_hourly_dispatch(tools):
     """
     Generates a matrix of hourly dispatch plots for each time region
@@ -598,7 +598,7 @@ def graph_hourly_dispatch(tools):
     )
 
 
-@graph("curtailment", title="Average daily curtailment")
+@graph("curtailment", title="Average daily curtailment", is_long=True)
 def graph_hourly_curtailment(tools):
     # Read dispatch.csv
     df = tools.get_dataframe("dispatch.csv")
@@ -615,6 +615,7 @@ def graph_hourly_curtailment(tools):
     "dispatch_per_scenario",
     title="Average daily dispatch",
     requires_multi_scenario=True,
+    is_long=True,
 )
 def graph_hourly_dispatch(tools):
     """
@@ -634,6 +635,7 @@ def graph_hourly_dispatch(tools):
     "curtailment_compare_scenarios",
     title="Average daily curtailment by scenario",
     requires_multi_scenario=True,
+    is_long=True,
 )
 def graph_hourly_curtailment(tools):
     # Read dispatch.csv
@@ -646,7 +648,11 @@ def graph_hourly_curtailment(tools):
     )
 
 
-@graph("total_dispatch", title="Total dispatched electricity")
+@graph(
+    "total_dispatch",
+    title="Total dispatched electricity",
+    is_long=True,
+)
 def graph_total_dispatch(tools):
     # ---------------------------------- #
     # total_dispatch.png                 #

switch_model/tools/graph/main.py

@@ -628,7 +628,7 @@ def save_figure(self, fig, filename=None):
 
     def run_graph_func(self, func):
         """Runs the graphing function"""
-        print(f"{func.name}", end=", ")
+        print(f"{func.name}", end=", ", flush=True)
         self._run_per_scenario = len(self._scenarios) > 1 and not func.multi_scenario
         self._figure_handler.set_properties(
             func.name,