diff --git a/homework_6/requirements.txt b/homework_6/requirements.txt index 3fa626f..a77d646 100644 --- a/homework_6/requirements.txt +++ b/homework_6/requirements.txt @@ -4,4 +4,4 @@ transformers==4.44.1 pytest==8.3.0 torch==2.2.1 great-expectations==0.15.25 - +pyarrow==17.0.0 diff --git a/homework_6/tests/data/test_data.py b/homework_6/tests/data/test_data.py index 0664d69..056f896 100644 --- a/homework_6/tests/data/test_data.py +++ b/homework_6/tests/data/test_data.py @@ -7,7 +7,7 @@ @pytest.fixture def dataset(): - df = pd.read_parquet("homework_6/pr2/data.parquet") + df = pd.read_parquet("pr2/data.parquet") return ge.dataset.PandasDataset(df)