Update src/skgmsh/__init__.py

pyvista · Mar 2, 2024 · a9e4eaa · a9e4eaa
1 parent f4eb665
commit a9e4eaa
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/src/skgmsh/__init__.py b/src/skgmsh/__init__.py
@@ -308,9 +308,9 @@ def __init__(
         )
 
 
-class Plotter(pv.Plotter):  # type: ignore[misc]
+class PlotterBase:
     """
-    Plotting object to display vtk meshes or numpy arrays.
+    Base class with common behaviour for a gmsh aware plotter.
 
     See :class:`pyvista.Plotter`.
 
@@ -328,7 +328,7 @@ class Plotter(pv.Plotter):  # type: ignore[misc]
 
     """
 
-    def __init__(self: Plotter, *args, **kwargs) -> None:  # type: ignore[no-untyped-def]  # noqa: ANN002, ANN003
+    def __init__(self: PlotterBase, *args, **kwargs) -> None:  # type: ignore[no-untyped-def]  # noqa: ANN002, ANN003
         """
         Create gmsh aware plotter.
 
@@ -346,7 +346,11 @@ def __init__(self: Plotter, *args, **kwargs) -> None:  # type: ignore[no-untyped
 
         """
         super().__init__(*args, **kwargs)
-        super().enable_parallel_projection()
+        super().enable_parallel_projection()  # type: ignore[misc]
+
+
+class Plotter(PlotterBase, pv.Plotter):  # type: ignore[misc]
+    """Plotting object to display vtk meshes or numpy arrays."""
 
 
 class PolyData(pv.PolyData):  # type: ignore[misc]