fix graph imports for performance tests

Signed-off-by: Thijs Baaijen <[email protected]>
PowerGridModel · Feb 3, 2025 · f7e0085 · f7e0085
1 parent a40bedd
commit f7e0085
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/tests/performance/_constants.py b/tests/performance/_constants.py
@@ -14,9 +14,9 @@
 }
 
 GRAPH_SETUP_CODES = {
-    "rustworkx": "from power_grid_model_ds.model.grids.base import Grid;"
-    + "from power_grid_model_ds.data_source.generator.grid_generators import RadialGridGenerator;"
-    + "from power_grid_model_ds.model.graphs.models import RustworkxGraphModel;"
+    "rustworkx": "from power_grid_model_ds import Grid;"
+    + "from power_grid_model_ds.generators import RadialGridGenerator;"
+    + "from power_grid_model_ds.graph_models import RustworkxGraphModel;"
     + "grid=RadialGridGenerator(nr_nodes={graph_size}, grid_class=Grid, graph_model=RustworkxGraphModel).run()",
 }
 

diff --git a/tests/performance/graph_performance_tests.py b/tests/performance/graph_performance_tests.py
@@ -48,9 +48,9 @@ def perftest_add_node():
 
 
 if __name__ == "__main__":
-    perftest_initialize()
-    perftest_set_feeder_ids()
+    # perftest_initialize()
+    # perftest_set_feeder_ids()
     perftest_get_components()
-    perftest_delete_node()
-    perftest_add_node()
-    perftest_from_arrays()
+    # perftest_delete_node()
+    # perftest_add_node()
+    # perftest_from_arrays()