From c6a774847523ee6cf05c0e7fe738cb8a53cd6691 Mon Sep 17 00:00:00 2001 From: Anatoly Myachev Date: Fri, 19 Apr 2024 16:44:17 +0200 Subject: [PATCH] fix Signed-off-by: Anatoly Myachev --- modin/tests/pandas/utils.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/modin/tests/pandas/utils.py b/modin/tests/pandas/utils.py index f34ff0e7c94..551bcc5cb80 100644 --- a/modin/tests/pandas/utils.py +++ b/modin/tests/pandas/utils.py @@ -1088,9 +1088,9 @@ def eval_io_from_str(csv_str: str, unique_filename: str, **kwargs): def create_test_dfs(*args, **kwargs) -> tuple[pd.DataFrame, pandas.DataFrame]: post_fn = kwargs.pop("post_fn", lambda df: df) - post_fn2 = lambda df: post_fn(df).convert_dtypes( - dtype_backend="dtype_backend" - ) # noqa: E731 + post_fn2 = lambda df: post_fn(df).convert_dtypes( # noqa: E731 + dtype_backend="pyarrow" + ) return tuple( map( post_fn2, [pd.DataFrame(*args, **kwargs), pandas.DataFrame(*args, **kwargs)] @@ -1109,8 +1109,8 @@ def create_test_series(vals, sort=False, **kwargs) -> tuple[pd.Series, pandas.Se modin_series = modin_series.sort_values().reset_index(drop=True) pandas_series = pandas_series.sort_values().reset_index(drop=True) - modin_series = modin_series.convert_dtypes(dtype_backend="dtype_backend") - pandas_series = pandas_series.convert_dtypes(dtype_backend="dtype_backend") + modin_series = modin_series.convert_dtypes(dtype_backend="pyarrow") + pandas_series = pandas_series.convert_dtypes(dtype_backend="pyarrow") return modin_series, pandas_series