Import narwhals' stable v1 API

Quantco · Sep 30, 2024 · c53d490 · c53d490
1 parent 267c321
commit c53d490
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/src/tabmat/categorical_matrix.py b/src/tabmat/categorical_matrix.py
@@ -167,7 +167,7 @@ def matvec(mat, vec):
 import warnings
 from typing import Optional, Union
 
-import narwhals as nw
+import narwhals.stable.v1 as nw
 import numpy as np
 from scipy import sparse as sps
 

diff --git a/src/tabmat/constructor.py b/src/tabmat/constructor.py
@@ -3,7 +3,7 @@
 from collections.abc import Mapping
 from typing import Any, Optional, Union
 
-import narwhals as nw
+import narwhals.stable.v1 as nw
 import numpy as np
 from formulaic import Formula, ModelSpec
 from formulaic.materializers.types import NAAction
@@ -156,7 +156,7 @@ def from_df(
                 dense_dfidx.append(dfcolidx)
                 dense_tmidx.append(mxcolidx)
                 mxcolidx += 1
-        elif isinstance(coldata.dtype, nw.dtypes.NumericType):
+        elif coldata.dtype.is_numeric():
             if (coldata != 0).mean() <= sparse_threshold:
                 sparse_dfidx.append(dfcolidx)
                 sparse_tmidx.append(mxcolidx)

diff --git a/tests/test_categorical_matrix.py b/tests/test_categorical_matrix.py
@@ -1,6 +1,6 @@
 import re
 
-import narwhals as nw
+import narwhals.stable.v1 as nw
 import numpy as np
 import pandas as pd
 import polars as pl