diff --git a/requirements.txt b/requirements.txt index 18c9532..233bd82 100644 --- a/requirements.txt +++ b/requirements.txt @@ -6,7 +6,7 @@ itsdangerous==2.0.1 gunicorn==20.0.4 model-archiver==1.0.3 multi-model-server==1.1.1 -numpy==1.21.0 +numpy==1.19.2 pandas==1.1.3 protobuf==3.20.1 psutil==5.7.2 diff --git a/test/unit/test_modules.py b/test/unit/test_modules.py index 1277993..41acffa 100644 --- a/test/unit/test_modules.py +++ b/test/unit/test_modules.py @@ -1,4 +1,12 @@ +import pandas as pd + + def test_pandas_version(): import pandas as pd major, minor, patch = pd.__version__.split('.') assert major == '1' + + +def test_pyarrow_to_parquet_conversion_regression_issue_106(): + df = pd.DataFrame({'x': [1, 2]}) + df.to_parquet('test.parquet', engine='pyarrow')