typing exts

narwhals-dev · Feb 19, 2024 · e455348 · e455348
1 parent a16ac9b
commit e455348
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/puffin/spec/__init__.py b/puffin/spec/__init__.py
@@ -1,12 +1,14 @@
 from __future__ import annotations
 
+from typing import TYPE_CHECKING
 from typing import Any
 from typing import Iterable
 from typing import Literal
 from typing import Protocol
 from typing import TypeVar
 
-from typing_extensions import Self
+if TYPE_CHECKING:
+    from typing_extensions import Self
 
 
 class Expr(Protocol):

diff --git a/t.py b/t.py
@@ -1,13 +1,12 @@
 # ruff: noqa
 import polars
 import pandas as pd
-import modin.pandas as mpd
 
 import puffin
 
-df_raw = mpd.DataFrame({"a": [1, 3, 2], "b": [4, 4, 6]})
+df_raw = pd.DataFrame({"a": [1, 3, 2], "b": [4, 4, 6]})
 df, pl = puffin.to_polars_api(df_raw, version="0.20")
-df_raw_2 = mpd.DataFrame({"a": [1, 3], "c": [7, 9]})
+df_raw_2 = pd.DataFrame({"a": [1, 3], "c": [7, 9]})
 df2, pl = puffin.to_polars_api(df_raw_2, version="0.20")
 
 result = df.sort("a", "b")