Skip to content

Commit

Permalink
fix geoplotter actor return typing (#439)
Browse files Browse the repository at this point in the history
  • Loading branch information
bjlittle authored Sep 16, 2023
1 parent fe39420 commit dc4c195
Showing 1 changed file with 6 additions and 7 deletions.
13 changes: 6 additions & 7 deletions src/geovista/geoplotter.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,6 @@

from pyproj import CRS
import pyvista as pv
import vtk

from .common import (
GV_FIELD_ZSCALE,
Expand Down Expand Up @@ -257,7 +256,7 @@ def _warn_opacity(self) -> None:

def add_base_layer(
self, mesh: pv.PolyData | None = None, **kwargs: Any | None
) -> vtk.vtkActor:
) -> pv.Actor:
"""Generate a cubed-sphere base layer mesh and add to the plotter scene.
Optionally, a `mesh` may be provided, which better fits the
Expand Down Expand Up @@ -289,7 +288,7 @@ def add_base_layer(
Returns
-------
vtkActor
Actor
The rendered actor added to the plotter scene.
Notes
Expand Down Expand Up @@ -347,7 +346,7 @@ def add_coastlines(
rtol: float | None = None,
atol: float | None = None,
**kwargs: Any | None,
) -> vtk.vtkActor:
) -> pv.Actor:
"""Generate coastlines and add to the plotter scene.
Parameters
Expand Down Expand Up @@ -375,7 +374,7 @@ def add_coastlines(
Returns
-------
vtkActor
Actor
The rendered actor added to the plotter scene.
Notes
Expand Down Expand Up @@ -509,7 +508,7 @@ def add_graticule(
point_labels_args=point_labels_args,
)

def add_mesh(self, mesh: Any, **kwargs: Any | None):
def add_mesh(self, mesh: Any, **kwargs: Any | None) -> pv.Actor:
"""Add the ``mesh`` to the plotter scene.
See :meth:`pyvista.Plotter.add_mesh`.
Expand Down Expand Up @@ -539,7 +538,7 @@ def add_mesh(self, mesh: Any, **kwargs: Any | None):
Returns
-------
vtkActor
Actor
The rendered actor added to the plotter scene.
Notes
Expand Down

0 comments on commit dc4c195

Please sign in to comment.