From 766ad777de1a80bb95ae911975901aac8859e30e Mon Sep 17 00:00:00 2001 From: Nikhil Raverkar Date: Thu, 14 Jul 2022 18:42:27 -0400 Subject: [PATCH] reverting numpy to 1.19.2 to solve pyarrow integration issue and adding regression test --- requirements.txt | 2 +- test/unit/test_modules.py | 8 ++++++++ 2 files changed, 9 insertions(+), 1 deletion(-) 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')