From 62c6cde1c7e2ec494a5a4d62bcbd71d0345c5b82 Mon Sep 17 00:00:00 2001
From: Thijs Baaijen <13253091+Thijss@users.noreply.github.com>
Date: Mon, 3 Feb 2025 21:26:22 +0100
Subject: [PATCH] fix constants for graph performance tests

Signed-off-by: Thijs Baaijen <13253091+Thijss@users.noreply.github.com>
---
 tests/performance/_constants.py              | 6 +++---
 tests/performance/graph_performance_tests.py | 1 -
 2 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/tests/performance/_constants.py b/tests/performance/_constants.py
index f114754..98c2754 100644
--- 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
index 8b242df..55279a4 100644
--- a/tests/performance/graph_performance_tests.py
+++ b/tests/performance/graph_performance_tests.py
@@ -7,7 +7,6 @@
 # pylint: disable=missing-function-docstring
 
 GRAPH_SIZES = [100, 500, 1000, 5000]
-GRAPH_SIZES = [100, 500, 1000]
 
 
 def perftest_initialize():