From 4a5d33416693c487a4ef05d76cfa4dc3d5b0af74 Mon Sep 17 00:00:00 2001 From: Filipp Shpomer Date: Fri, 22 May 2026 16:38:39 -0700 Subject: [PATCH] fix: remove unsupported dd.DataFrame type in parquet_reader --- cdisc_rules_engine/services/data_readers/parquet_reader.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/cdisc_rules_engine/services/data_readers/parquet_reader.py b/cdisc_rules_engine/services/data_readers/parquet_reader.py index 1df298969..953c246d4 100644 --- a/cdisc_rules_engine/services/data_readers/parquet_reader.py +++ b/cdisc_rules_engine/services/data_readers/parquet_reader.py @@ -1,5 +1,4 @@ from io import BytesIO -from typing import Union import pandas as pd import dask.dataframe as dd @@ -29,9 +28,7 @@ def from_file(self, file_path): file_path ) - def _format_floats( - self, dataframe: Union[pd.DataFrame, dd.DataFrame] - ) -> Union[pd.DataFrame, dd.DataFrame]: + def _format_floats(self, dataframe: pd.DataFrame) -> pd.DataFrame: return dataframe.map(lambda x: round(x, 15) if isinstance(x, float) else x) def _read_dask(self, file_path):