[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

cicd_utils/cicd/test_helpers.py

@@ -57,7 +57,7 @@ def round_trip_pickle(obj: _T, protocol: int = pickle.HIGHEST_PROTOCOL) -> _T:
     The object that was pickled and unpickled.
 
     """
-    return cast(_T, pickle.loads(pickle.dumps(obj, protocol=protocol)))  # noqa: S301
+    return cast("_T", pickle.loads(pickle.dumps(obj, protocol=protocol)))  # noqa: S301
 
 
 def import_pyscript_as_module(path: str | Path) -> ModuleType:
@@ -81,15 +81,15 @@ def import_pyscript_as_module(path: str | Path) -> ModuleType:
     path_posix = Path(path).resolve().as_posix()
     module_name = path_posix.split("/")[-1].split(".")[0]
     spec = cast(
-        ModuleSpec,
+        "ModuleSpec",
         spec_from_file_location(
             name=module_name,
             location=path_posix,
         ),
     )
     module = module_from_spec(spec)
     sys.modules[module_name] = module
-    loader = cast(Loader, spec.loader)
+    loader = cast("Loader", spec.loader)
     loader.exec_module(module)
     return module
 

src/ridgeplot/_color/colorscale.py

@@ -23,7 +23,7 @@ def __init__(self) -> None:
     @property
     @override
     def named_colorscales(self) -> dict[str, list[str]]:
-        named_colorscales = cast(dict[str, list[str]], super().named_colorscales)
+        named_colorscales = cast("dict[str, list[str]]", super().named_colorscales)
         if "default" not in named_colorscales:
             # Add 'default' for backwards compatibility
             named_colorscales["default"] = px.colors.DEFAULT_PLOTLY_COLORS
@@ -34,12 +34,12 @@ def validate_coerce(self, v: Any) -> ColorScale:
         coerced = super().validate_coerce(v)
         if coerced is None:  # pragma: no cover
             self.raise_invalid_val(coerced)
-        coerced = cast(ColorScale, coerced)
+        coerced = cast("ColorScale", coerced)
         # This helps us avoid floating point errors when making
         # comparisons in our test suite. The user should not
         # be able to notice *any* difference in the output
         coerced = tuple((v if isinstance(v, int) else round(v, ndigits=12), c) for v, c in coerced)
-        return cast(ColorScale, coerced)
+        return cast("ColorScale", coerced)
 
 
 def infer_default_colorscale() -> ColorScale | Collection[Color] | str:

src/ridgeplot/_color/utils.py

@@ -21,7 +21,7 @@ def default_plotly_template() -> go.layout.Template:
 #       once we add support for color sequences.
 def infer_default_color_sequence() -> Collection[Color]:  # pragma: no cover
     return cast(
-        Collection[Color], default_plotly_template().layout.colorway or px.colors.qualitative.D3
+        "Collection[Color]", default_plotly_template().layout.colorway or px.colors.qualitative.D3
     )
 
 
@@ -32,7 +32,7 @@ def to_rgb(color: Color) -> str:
         r, g, b = color
         rgb = f"rgb({r}, {g}, {b})"
     elif color.startswith("#"):
-        return to_rgb(cast(str, px.colors.hex_to_rgb(color)))
+        return to_rgb(cast("str", px.colors.hex_to_rgb(color)))
     elif color.startswith(("rgb(", "rgba(")):
         rgb = color
     elif color in CSS_NAMED_COLORS:
@@ -50,7 +50,7 @@ def unpack_rgb(rgb: str) -> tuple[float, float, float, float] | tuple[float, flo
     prefix = rgb.split("(")[0] + "("
     values_str = map(str.strip, rgb.removeprefix(prefix).removesuffix(")").split(","))
     values_num = tuple(int(v) if v.isdecimal() else float(v) for v in values_str)
-    return cast(Union[tuple[float, float, float, float], tuple[float, float, float]], values_num)
+    return cast("Union[tuple[float, float, float, float], tuple[float, float, float]]", values_num)
 
 
 def apply_alpha(color: Color, alpha: float) -> str:

src/ridgeplot/_figure_factory.py

@@ -48,7 +48,7 @@ def normalise_trace_types(
     trace_types: TraceTypesArray | ShallowTraceTypesArray | TraceType,
 ) -> TraceTypesArray:
     if is_trace_type(trace_types):
-        trace_types = cast(TraceTypesArray, [[trace_types] * len(row) for row in densities])
+        trace_types = cast("TraceTypesArray", [[trace_types] * len(row) for row in densities])
     elif is_shallow_trace_types_array(trace_types):
         trace_types = nest_shallow_collection(trace_types)
         trace_types = normalise_row_attrs(trace_types, l2_target=densities)

src/ridgeplot/_kde.py

@@ -183,7 +183,7 @@ def _validate_densities(
         # Fail early if the return type is incorrect
         # Otherwise, the remaining checks will fail
         raise RuntimeError(msg)  # noqa: TRY004
-    y = cast(npt.NDArray[np.floating[Any]], y)
+    y = cast("npt.NDArray[np.floating[Any]]", y)
     wrong_shape = y.shape != x.shape
     not_finite = ~np.isfinite(y).all()
     if wrong_shape or not_finite:

src/ridgeplot/_ridgeplot.py

@@ -78,7 +78,7 @@ def _coerce_to_densities(
     # Transform samples into densities via KDE or histogram binning
     if is_shallow_samples(samples):
         samples = nest_shallow_collection(samples)
-    samples = cast(Samples, samples)
+    samples = cast("Samples", samples)
     if nbins is not None:
         densities = bin_samples(
             samples=samples,

tests/unit/conftest.py

@@ -48,7 +48,7 @@ def viridis_colorscale() -> ColorScale:
 def valid_colorscale(
     request: pytest.FixtureRequest,
 ) -> tuple[ColorScale | Collection[Color] | str, ColorScale]:
-    return cast(tuple[Union[ColorScale, Collection[Color], str], ColorScale], request.param)
+    return cast("tuple[Union[ColorScale, Collection[Color], str], ColorScale]", request.param)
 
 
 INVALID_COLOR_SCALES = [
@@ -66,4 +66,4 @@ def valid_colorscale(
 
 @pytest.fixture(scope="session", params=INVALID_COLOR_SCALES)
 def invalid_colorscale(request: pytest.FixtureRequest) -> ColorScale | Collection[Color] | str:
-    return cast(Union[ColorScale, Collection[Color], str], request.param)
+    return cast("Union[ColorScale, Collection[Color], str]", request.param)