format

Author: aoustry

src/andromede/pypsa_converter/pypsa_converter.py

@@ -162,6 +162,7 @@ def _pypsa_network_assertion(self) -> None:
                 ]
                 == "co2_emissions"
             )
+
     def _rename_buses(self) -> None:    
         ### Rename PyPSA buses, to delete spaces
         if len(self.pypsa_network.buses) > 0:
@@ -177,7 +178,8 @@ def _rename_buses(self) -> None:
                 for col in ["bus", "bus0", "bus1"]:
                     if col in df.columns:
                         df[col] = df[col].str.replace(" ", "_")
-                        
+
+
     def _pypsa_network_preprocessing(self) -> None:
         ###Add fictitious carrier
         self.pypsa_network.add(
@@ -191,9 +193,9 @@ def _pypsa_network_preprocessing(self) -> None:
         ] = self.pypsa_network.carriers.index.values
         self._rename_buses()
 
-    def _rename_pypsa_components(self, component_type: str)-> None:
+    def _rename_pypsa_components(self, component_type: str) -> None:
         df = getattr(self.pypsa_network, component_type)
-        if len(df)==0:
+        if len(df) == 0:
             return
         ### Rename PyPSA components, to make sure that the names are uniques (used as id in the Gems model)
         prefix = component_type[:-1]
@@ -202,9 +204,9 @@ def _rename_pypsa_components(self, component_type: str)-> None:
         for _, val in dictionnary.items():
             val.columns = prefix + "_" + val.columns.str.replace(" ", "_")
 
-    def _fix_capacities(self, component_type: str, capa_str:str)-> None:
+    def _fix_capacities(self, component_type: str, capa_str:str) -> None:
         df = getattr(self.pypsa_network, component_type)
-        if len(df)==0:
+        if len(df) == 0:
             return
         ### Adding min and max capacities to non-extendable objects
         for field in [capa_str + "_min", capa_str + "_max"]:
@@ -231,7 +233,7 @@ def _preprocess_pypsa_components(
         )
         self._rename_pypsa_components(component_type)
         if extendable:
-            self._fix_capacities(component_type,capa_str)
+            self._fix_capacities(component_type, capa_str)
 
 
     def _register_pypsa_components(self) -> None:

tests/pypsa_converter/test_advanced_pypsa_cases.py

@@ -211,4 +211,3 @@ def test_case_gemspy() -> None:
     pypsa_gemspy_benchmark("base_s_6_elec_lvopt_.nc", 0.4, True)
     pypsa_gemspy_benchmark("base_s_6_elec_lvopt_.nc", 0.3, True)
     pypsa_gemspy_benchmark("simple.nc", 1.0, False)
-