Fix for 'import error' in merlin-tensorflow-training:22.05 (#317)

NVIDIA-Merlin · May 17, 2022 · 57650bd · 57650bd
1 parent 56de5d4
commit 57650bd
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/docker/training/dockerfile.tf b/docker/training/dockerfile.tf
@@ -104,6 +104,7 @@ RUN if [ "$HUGECTR_DEV_MODE" == "false" ]; then \
         popd && \
         rm -rf build-env; \
     fi
+ENV PYTHONPATH=$PYTHONPATH:/usr/lib/python3.8/site-packages/merlin_sok-1.1.3-py3.8-linux-x86_64.egg
 
 # Install distributed-embeddings
 ARG INSTALL_DISTRIBUTED_EMBEDDINGS=true