Skip to content

Commit 7ef6e76

Browse files
committed
fix test
1 parent 22303cb commit 7ef6e76

1 file changed

Lines changed: 1 addition & 2 deletions

File tree

tests/test_estimators.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,4 @@
11
import pytest
2-
import narwhals as nw
32

43
from playtime.estimators import SelectCols
54

@@ -8,7 +7,7 @@
87
def test_select_cols(lib):
98
lib = pytest.importorskip(lib)
109

11-
df = nw.from_native(lib.DataFrame({"a": [1, 2], "b": [10, 20], "c": ["x", "y"]}))
10+
df = lib.DataFrame({"a": [1, 2], "b": [10, 20], "c": ["x", "y"]})
1211
tfm = SelectCols(["a", "b"])
1312
# This should work w/o calling `fit`
1413
assert tfm.transform(df).columns == ["a", "b"]

0 commit comments

Comments
 (0)