From d241d546b702f129ec687be58417075b326803da Mon Sep 17 00:00:00 2001
From: Julio Perez <37191411+jperez999@users.noreply.github.com>
Date: Wed, 18 May 2022 13:36:03 -0400
Subject: [PATCH] Distributed embeddings failing to install because of
 PYTHONPATH Edit before hand (#326)

* add entrypoint to all containers

* remove -e for pip installs

* add keyring to fix bad key nvidia issue

* trying to move env call to see if it helps fix the install of distributed embeddings after sok install
---
 docker/training/dockerfile.tf | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/docker/training/dockerfile.tf b/docker/training/dockerfile.tf
index 5f6a4614f..053f83257 100644
--- a/docker/training/dockerfile.tf
+++ b/docker/training/dockerfile.tf
@@ -104,7 +104,6 @@ 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
@@ -114,6 +113,9 @@ RUN if [ "$INSTALL_DISTRIBUTED_EMBEDDINGS" == "true" ]; then \
         make pip_pkg && pip install artifacts/*.whl && make clean; \
     fi
 
+ENV PYTHONPATH=$PYTHONPATH:/usr/lib/python3.8/site-packages/merlin_sok-1.1.3-py3.8-linux-x86_64.egg
+
+
 # Clean up
 RUN rm -rf /repos
 RUN rm -rf /usr/local/share/jupyter/lab/staging/node_modules/marked