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

@@ -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