From 670ff5e09c81d8c622d4ce3a0dae8f69dc820754 Mon Sep 17 00:00:00 2001
From: CarlosEpia <carlosepiar@gmail.com>
Date: Mon, 7 Aug 2023 15:33:03 +0200
Subject: [PATCH] add plot_carrier to the etrago methods

---
 etrago/tools/network.py | 3 +++
 etrago/tools/plot.py    | 4 ++--
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/etrago/tools/network.py b/etrago/tools/network.py
index 97ffbfe77..732ced23b 100644
--- 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
index 89573506d..707092ec2 100644
--- 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"