Skip to content

Commit b0b310c

Browse files
committed
more files
Signed-off-by: Thijs Baaijen <[email protected]>
1 parent 177f9cb commit b0b310c

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

src/power_grid_model_ds/_core/data_source/generator/grid_generators.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -75,6 +75,6 @@ def run(self, seed=None, create_10_3_kv_net: bool = False) -> T:
7575
grid.append(transformers)
7676

7777
lines = line_generator.run(amount=0, number_of_routes=0)
78-
grid.append(lines[~np.isin(lines.id, grid.line.id)])
78+
grid.append(lines[~np.isin(lines["id"], grid.line["id"])])
7979

8080
return grid

tests/performance/grid_performance_tests.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -28,8 +28,8 @@ def perf_test_add_lines():
2828
+ "nodes = NodeArray.zeros({size});"
2929
+ "grid.append(nodes);"
3030
+ "lines = LineArray.zeros({size});"
31-
+ "lines.from_node = nodes.id;"
32-
+ "lines.to_node = nodes.id;"
31+
+ "lines['from_node'] = nodes['id'];"
32+
+ "lines['to_node'] = nodes['id'];"
3333
}
3434

3535
code_to_test = ["grid.append(lines);"]
@@ -45,7 +45,7 @@ def perf_test_get_downstream_nodes_performance():
4545
+ "from power_grid_model_ds.generators import RadialGridGenerator;"
4646
+ "from power_grid_model_ds.graph_models import RustworkxGraphModel;"
4747
+ "grid=RadialGridGenerator(nr_nodes={size}, grid_class=Grid, graph_model=RustworkxGraphModel).run();"
48-
+ "non_substation_node = grid.node.filter(node_type=NodeType.UNSPECIFIED).id;"
48+
+ "non_substation_node = grid.node.filter(node_type=NodeType.UNSPECIFIED)['id'];"
4949
+ "node_id = np.random.choice(non_substation_node)"
5050
}
5151

0 commit comments

Comments
 (0)