Fix connect_to_lv_bus return type and error handling

openego · Sep 27, 2024 · 0c8b175 · 0c8b175
1 parent b2d6cff
commit 0c8b175
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/edisgo/network/topology.py b/edisgo/network/topology.py
@@ -2603,7 +2603,10 @@ def handle_voltage_level_7():
                 target_bus = self._connect_to_lv_bus(
                     edisgo_object, lv_buses.distance.idxmin(), comp_type, comp_data
                 )
-                return target_bus, ModuleNotFoundError
+                if isinstance(target_bus, str):
+                    return target_bus, None
+                elif isinstance(target_bus, pd.DataFrame):
+                    return target_bus.index[0], None
 
             else:
                 target_bus = lv_buses_masked.loc[lv_buses_masked.distance.idxmin()]