add plot_carrier to the etrago methods

openego · Aug 7, 2023 · 670ff5e · 670ff5e
1 parent 6e2a0d5
commit 670ff5e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/etrago/tools/network.py b/etrago/tools/network.py
@@ -54,6 +54,7 @@
     heat_stores,
     hydrogen_stores,
     plot_clusters,
+    plot_carrier,
     plot_gas_generation,
     plot_gas_summary,
     plot_grid,
@@ -267,6 +268,8 @@ def __init__(
     plot_grid = plot_grid
 
     plot_clusters = plot_clusters
+
+    plot_carrier = plot_carrier
 
     plot_gas_generation = plot_gas_generation
 

diff --git a/etrago/tools/plot.py b/etrago/tools/plot.py
@@ -2134,7 +2134,7 @@ def flexibility_usage(
         fig_e.savefig(pre_path + f"stored_e_{flexibility}")
 
 
-def plot_carrier(network, carrier_links=["AC"], carrier_buses=["AC"]):
+def plot_carrier(etrago, carrier_links=["AC"], carrier_buses=["AC"]):
     """
     Parameters
     ----------
@@ -2152,7 +2152,7 @@ def plot_carrier(network, carrier_links=["AC"], carrier_buses=["AC"]):
     None.
 
     """
-
+    network = etrago.network
     colors = coloring()
     line_colors = "lightblue"