Skip to content

Commit d73d939

Browse files
committed
fix mypy error
1 parent 0f152b2 commit d73d939

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

netcbs/netcbs.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44

55
import logging
66
from pathlib import Path
7-
from typing import Dict, List, Set, Tuple, Union, Sequence
7+
from typing import Dict, List, Set, Tuple, Union, Sequence, cast
88

99
import duckdb
1010
import pandas as pd
@@ -453,7 +453,7 @@ def transform(
453453
ON result.RINPERSOON = sample.RINPERSOON
454454
""").fetch_arrow_table()
455455

456-
df_res = pl.from_arrow(arrow_res).select(
457-
pl.all().exclude(["RINPERSOONsample", "RINPERSOON_1"])
458-
)
456+
df_res = cast(pl.DataFrame, pl.from_arrow(arrow_res))
457+
df_res = df_res.select(pl.all().exclude(["RINPERSOONsample", "RINPERSOON_1"]))
458+
459459
return df_res.to_pandas() if return_pandas else df_res

0 commit comments

Comments
 (0)