fix typing

Signed-off-by: Vincent Koppen <[email protected]>

Author: vincentkoppen

src/power_grid_model_ds/_core/model/graphs/models/base.py

@@ -183,7 +183,7 @@ def delete_branch3_array(self, branch_array: Branch3Array, raise_on_fail: bool =
             self.delete_branch_array(branches, raise_on_fail=raise_on_fail)
 
     @contextmanager
-    def tmp_remove_nodes(self, nodes: list[int]) -> Generator:
+    def tmp_remove_nodes(self, nodes: list[int] | NDArray[np.int32]) -> Generator:
         """Context manager that temporarily removes nodes and their branches from the graph.
         Example:
             >>> with graph.tmp_remove_nodes([1, 2, 3]):

src/power_grid_model_ds/_core/model/grids/helpers.py

@@ -45,7 +45,7 @@ def set_feeder_ids(grid: "Grid"):
     601 | 101          | 204
     """
     _reset_feeder_ids(grid)
-    feeder_node_ids = grid.node.filter(node_type=NodeType.SUBSTATION_NODE).id
+    feeder_node_ids = grid.node.filter(node_type=NodeType.SUBSTATION_NODE)["id"]
     components = grid.graphs.active_graph.get_components(feeder_node_ids)
     for component_node_ids in components:
         component_branches = _get_active_component_branches(grid, component_node_ids)