setup.py fixes

mlfoundations · Dec 12, 2023 · a9f9d5c · a9f9d5c
1 parent acec8b3
commit a9f9d5c
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/sagemaker_train/.dockerignore b/sagemaker_train/.dockerignore
@@ -1,4 +0,0 @@
-setup.py
-.setup.py
-./../setup.py
-../setup.py

diff --git a/sagemaker_train/Dockerfile b/sagemaker_train/Dockerfile
@@ -21,6 +21,7 @@ ENV SAGEMAKER_SUBMIT_DIRECTORY /opt/ml/code
 
 # /opt/ml and all subdirectories are utilized by SageMaker, use the /code subdirectory to store your user code.
 COPY . /opt/ml/code/
+RUN rm /opt/ml/code/setup.py
 
 RUN pip install -r /opt/ml/code/requirements.txt
 RUN pip uninstall flash-attn -y