cleanup code

Signed-off-by: Thijs Baaijen <[email protected]>

Author: Thijss

src/power_grid_model_ds/visualizer.py

@@ -6,6 +6,7 @@
     from power_grid_model_ds._core.visualizer.main import visualize
 except ImportError as error:
     raise ImportError(
-        "Missing dependencies for visualizer: install with 'pip install power-grid-model-ds[visualizer]'") from error
+        "Missing dependencies for visualizer: install with 'pip install power-grid-model-ds[visualizer]'"
+    ) from error
 
 __all__ = ["visualize"]

tests/integration/visualizer_tests.py

@@ -7,7 +7,8 @@
 import numpy as np
 from numpy.typing import NDArray
 
-from power_grid_model_ds import Grid, visualize
+from power_grid_model_ds import Grid
+from power_grid_model_ds._core.visualizer.main import visualize
 from power_grid_model_ds.arrays import NodeArray
 from power_grid_model_ds.generators import RadialGridGenerator