From a98870e1597cc5cd25daa1d11cae14125960edec Mon Sep 17 00:00:00 2001
From: Fabian Neumann <fabian.neumann@outlook.de>
Date: Wed, 7 Jul 2021 18:07:57 +0200
Subject: [PATCH] correction for .madd() and call co2_network

---
 scripts/prepare_sector_network.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/scripts/prepare_sector_network.py b/scripts/prepare_sector_network.py
index aca8d76f..c5a99fcc 100644
--- a/scripts/prepare_sector_network.py
+++ b/scripts/prepare_sector_network.py
@@ -284,9 +284,9 @@ def add_co2_tracking(n, options):
         co2_vents = nodes + " co2 vent"
         co2_max_sequestration = np.inf
     else:
-        co2_nodes = "co2 stored"
+        co2_nodes = ["co2 stored"]
         co2_locations = "EU"
-        co2_vents = "co2 vent"
+        co2_vents = ["co2 vent"]
         co2_max_sequestration = np.inf # TODO should be nodal sequestration potentials
 
     # minus sign because opposite to how fossil fuels used:
@@ -2074,6 +2074,9 @@ if __name__ == "__main__":
     if "noH2network" in opts:
         remove_h2_network(n)
 
+    if options["co2_network"]:
+        add_co2_network(n, costs)
+
     for o in opts:
         m = re.match(r'^\d+h$', o, re.IGNORECASE)
         if m is not None: