Merge pull request #124 from aws/issue_106

Reverting numpy 1.21.2 to 1.19.2 and adding regression
aws · Jul 15, 2022 · 980f6d2 · 980f6d2
2 parents 7773e19 + 766ad77
commit 980f6d2
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git 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
@@ -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')