From 46ed00a08ed3bdfab4aa7b7d7899d59e48af5a23 Mon Sep 17 00:00:00 2001 From: hirosassa Date: Wed, 13 May 2026 11:58:53 +0900 Subject: [PATCH] add casts for pandas type stub incompatibilities --- gokart/target.py | 2 +- gokart/utils.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/gokart/target.py b/gokart/target.py index 9792032d..85c02dde 100644 --- a/gokart/target.py +++ b/gokart/target.py @@ -226,7 +226,7 @@ def save(self, df: pd.DataFrame, file_path: str) -> None: def load(file_path: str) -> pd.DataFrame: dir_path = os.path.dirname(file_path) - return pd.concat([pd.read_pickle(fp) for fp in glob(os.path.join(dir_path, 'data_*.pkl'))]) + return cast(pd.DataFrame, pd.concat([pd.read_pickle(fp) for fp in glob(os.path.join(dir_path, 'data_*.pkl'))])) def _make_file_system_target(file_path: str, processor: FileProcessor | None = None, store_index_in_feather: bool = True) -> luigi.target.FileSystemTarget: diff --git a/gokart/utils.py b/gokart/utils.py index f0b19768..5f1ec6bc 100644 --- a/gokart/utils.py +++ b/gokart/utils.py @@ -3,7 +3,7 @@ import os from collections.abc import Callable, Iterable from io import BytesIO -from typing import Any, Literal, Protocol, TypeAlias, TypeVar, get_args, get_origin +from typing import Any, Literal, Protocol, TypeAlias, TypeVar, cast, get_args, get_origin import dill import luigi @@ -91,7 +91,7 @@ def load_dill_with_pandas_backward_compatibility(file: FileLike | BytesIO) -> An except Exception: assert file.seekable(), f'{file} is not seekable.' file.seek(0) - return pd.read_pickle(file) + return pd.read_pickle(cast(Any, file)) def get_dataframe_type_from_task(task: Any) -> Literal['pandas', 'polars', 'polars-lazy']: