Make type checking block

mcrumiller · Jan 12, 2025 · 20fa148 · 20fa148
1 parent 96ded25
commit 20fa148
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/py-polars/tests/unit/functions/range/test_int_range.py b/py-polars/tests/unit/functions/range/test_int_range.py
@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from typing import Any
+from typing import TYPE_CHECKING, Any
 
 import numpy as np
 import pytest
@@ -9,6 +9,9 @@
 from polars.exceptions import ComputeError, InvalidOperationError
 from polars.testing import assert_frame_equal, assert_series_equal
 
+if TYPE_CHECKING:
+    from polars._typing import ClosedInterval
+
 
 def test_int_range() -> None:
     result = pl.int_range(0, 3)
@@ -289,7 +292,7 @@ def test_linear_space(
     start: int | float,
     end: int | float,
     num_samples: int,
-    interval: str,
+    interval: ClosedInterval,
     eager: bool,
 ) -> None:
     if eager: