Skip to content

Commit

Permalink
reverting numpy to 1.19.2, updating boto3 to 1.24.17, fixing docker b…
Browse files Browse the repository at this point in the history
…uild issue and adding regression test
  • Loading branch information
Nikhil Raverkar committed Jul 15, 2022
1 parent f8015ce commit 2268f2f
Show file tree
Hide file tree
Showing 4 changed files with 14 additions and 5 deletions.
3 changes: 2 additions & 1 deletion docker/0.23-1/final/Dockerfile.cpu
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,8 @@ RUN python -m pip install -r /requirements.txt && \

COPY dist/sagemaker_sklearn_container-2.0-py3-none-any.whl /sagemaker_sklearn_container-2.0-py3-none-any.whl
# https://github.com/googleapis/google-cloud-python/issues/6647
RUN rm -rf /miniconda3/lib/python3.7/site-packages/numpy-1.19.4.dist-info && \
RUN conda install numpy --force-reinstall && \
rm -rf /miniconda3/lib/python3.7/site-packages/numpy-1.19.4.dist-info && \
pip install --no-cache /sagemaker_sklearn_container-2.0-py3-none-any.whl && \
rm /sagemaker_sklearn_container-2.0-py3-none-any.whl

Expand Down
6 changes: 3 additions & 3 deletions requirements.txt
Original file line number Diff line number Diff line change
@@ -1,12 +1,12 @@
boto3==1.16.4
botocore==1.19.4
boto3==1.24.17
botocore==1.27.18
cryptography==35.0.0
Flask==1.1.1
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
psutil==5.7.2
python-dateutil==2.8.1
Expand Down
2 changes: 1 addition & 1 deletion test-requirements.txt
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
Flask
PyYAML
boto3>=1.13.17
boto3>=1.24.17
coverage
docker-compose
flake8
Expand Down
8 changes: 8 additions & 0 deletions test/unit/test_modules.py
Original file line number Diff line number Diff line change
@@ -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')

0 comments on commit 2268f2f

Please sign in to comment.