Skip to content

Commit

Permalink
Old-style type hints
Browse files Browse the repository at this point in the history
  • Loading branch information
JBorrow committed Sep 11, 2024
1 parent aa58411 commit 6702112
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions swiftsimio/visualisation/volume_render.py
Original file line number Diff line number Diff line change
Expand Up @@ -882,11 +882,11 @@ def render_voxel_to_array(data, center, width):
def visualise_render(
render: ndarray,
centers: list[float],
widths: list[float] | float,
widths: Union[list[float], float],
cmap: str = "viridis",
return_type: Literal["all", "lighten", "add"] = "lighten",
norm: Union[list["plt.Normalize"], "plt.Normalize", None] = None,
) -> tuple[list[ndarray] | ndarray, list["plt.Normalize"]]:
) -> tuple[Union[list[ndarray], ndarray], list["plt.Normalize"]]:
"""
Visualises a render with multiple centers and widths.
Expand Down Expand Up @@ -961,7 +961,7 @@ def visualise_render(

def visualise_render_options(
centers: list[float],
widths: list[float] | float,
widths: Union[list[float], float],
cmap: str = "viridis",
) -> tuple["plt.Figure", "plt.Axes"]:
"""
Expand Down

0 comments on commit 6702112

Please sign in to comment.