issue #4 bump aoa client version to 2.8.0rc5 and fix shap install

ThinkBigAnalytics · Jan 15, 2021 · 94053fc · 94053fc
1 parent f1cb808
commit 94053fc
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/aoa-pyspark-base b/aoa-pyspark-base
@@ -8,6 +8,7 @@ RUN apt-get install -y zip openjdk-8-jdk && \
 
 ENV JAVA_HOME="/usr/lib/jvm/java-8-openjdk-amd64/"
 
-RUN conda create -n pyspark -c conda-forge python=3.6 pyspark=2.4.3 pandas=0.24.2 numpy=1.16.1 scikit-learn=0.20.3 conda-pack aoa==2.8.0rc5
+RUN conda create -n pyspark -c conda-forge python=3.6 pyspark=2.4.3 pandas=0.24.2 numpy=1.16.1 scikit-learn=0.20.3 conda-pack shap==0.37.0
+RUN /usr/local/envs/pyspark/bin/pip install teradataml==17.0.0.0 nyoka==4.3.0 xgboost==0.90 aoa==2.8.0rc5
 
 CMD ["bash"]
diff --git a/aoa-python-base b/aoa-python-base
@@ -4,6 +4,8 @@ RUN apt-get update -y --fix-missing
 RUN apt-get install -y python-pip python-dev build-essential gunicorn openjdk-8-jdk-headless && \
     apt-get clean
 
+RUN pip install --upgrade setuptools
+
 RUN pip install --no-cache-dir numpy==1.16.1 \
                tensorflow==1.8.0 \
                keras==2.2.0 \