move visualizer import

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

Author: Thijss

src/power_grid_model_ds/__init__.py

@@ -5,9 +5,5 @@
 from power_grid_model_ds._core.load_flow import PowerGridModelInterface
 from power_grid_model_ds._core.model.graphs.container import GraphContainer
 from power_grid_model_ds._core.model.grids.base import Grid
-try:
-    from power_grid_model_ds._core.visualizer.main import visualize
-except ImportError:
-    pass
 
-__all__ = ["Grid", "GraphContainer", "PowerGridModelInterface", "visualize"]
+__all__ = ["Grid", "GraphContainer", "PowerGridModelInterface"]

src/power_grid_model_ds/visualizer.py

@@ -0,0 +1,11 @@
+# SPDX-FileCopyrightText: Contributors to the Power Grid Model project <[email protected]>
+#
+# SPDX-License-Identifier: MPL-2.0
+
+try:
+    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
+
+__all__ = ["visualize"]