FROM preprod-sklearn:1.2-1

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

RUN pip install --upgrade --no-cache --no-deps sagemaker-scikit-learn-extension==2.5.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