Added workaround for alibi explainer image build

SeldonIO · Dec 11, 2020 · 9fb1910 · 9fb1910
1 parent e5bd308
commit 9fb1910
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/components/alibi-explain-server/Dockerfile b/components/alibi-explain-server/Dockerfile
@@ -12,6 +12,9 @@ COPY setup.py setup.py
 COPY alibiexplainer alibiexplainer
 COPY README.md README.md
 
+# Required for https://github.com/slundberg/shap/issues/1633
+RUN pip install numpy==1.19.04 
+
 RUN pip install . --no-binary protobuf
 
 RUN chmod -R a+rwx /opt/app-root/lib/python3.6

diff --git a/components/alibi-explain-server/alibiexplainer/proto/tensorflow/core/framework/types.proto b/components/alibi-explain-server/alibiexplainer/proto/tensorflow/core/framework/types.proto
@@ -84,4 +84,6 @@ enum SpecializedType {
   ST_INVALID = 0;
   // "tensorflow::TensorList" in the variant type registry.
   ST_TENSOR_LIST = 1;
-}
+  // "tensorflow::data::Optional" in the variant type registry.
+  ST_OPTIONAL = 2;
+}