Correct Dockerfile and requirements

datalab-mi · Mar 30, 2021 · 366f70a · 366f70a
1 parent 08f5a17
commit 366f70a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -1,7 +1,7 @@
 FROM ovhcom/ai-training-pytorch:1.6.0
 LABEL maintainer="datalab-mi"
 
-COPY requirements.txt ./
-RUN pip install -r -y --no-cache-dir requirements.txt
+COPY requirements.txt requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
 
 
diff --git a/requirements.txt b/requirements.txt
@@ -2,4 +2,4 @@ pandas>=1.2.0
 requests>=2.21.0
 pytest==5.1.1
 transformers>=3.1.0,<5.0.0
-sentence-transformers=1.0.3
+sentence-transformers==1.0.3