FROM preprod-sklearn:0.23-1-arm64-cpu-py37

RUN pip freeze | grep -q 'scikit-learn==0.23.2'; \
	if [ $? -eq 0 ]; \
		then echo 'scikit-learn version 0.23.2 requirement met'; \
		else echo 'ERROR: Expected scikit-learn version is 0.23.2, check base images for scikit-learn version' && \
			 exit 1; fi

RUN pip install --upgrade --no-cache --no-deps sagemaker-scikit-learn-extension==2.4.0
RUN python -m pip install statsmodels==0.9.0
RUN python -m pip install matrixprofile==1.1.10
RUN python -m pip install stumpy==1.7.2
RUN python -m pip install --no-deps tsfresh==0.17.0