Removed torchtext as the project has stopped releasing new torch comp…

…atible versions
pytorch · Aug 5, 2024 · 6bd7ee4 · 6bd7ee4
1 parent 6ce3f45
commit 6bd7ee4
Show file tree

Hide file tree

Showing 5 changed files with 4 additions and 9 deletions.
diff --git a/docker/hvd/Dockerfile.hvd-apex-nlp b/docker/hvd/Dockerfile.hvd-apex-nlp
@@ -4,5 +4,4 @@ FROM pytorchignite/hvd-apex:latest
 # Ignite NLP dependencies
 RUN pip install --upgrade --no-cache-dir transformers \
                                          spacy \
-                                         nltk \
-                                         torchtext
+                                         nltk
diff --git a/docker/hvd/Dockerfile.hvd-nlp b/docker/hvd/Dockerfile.hvd-nlp
@@ -4,5 +4,4 @@ FROM pytorchignite/hvd-base:latest
 # Ignite NLP dependencies
 RUN pip install --upgrade --no-cache-dir transformers \
                                          spacy \
-                                         nltk \
-                                         torchtext
+                                         nltk
diff --git a/docker/main/Dockerfile.apex-nlp b/docker/main/Dockerfile.apex-nlp
@@ -4,5 +4,4 @@ FROM pytorchignite/apex:latest
 # Ignite NLP dependencies
 RUN pip install --upgrade --no-cache-dir transformers \
                                          spacy \
-                                         nltk \
-                                         torchtext
+                                         nltk
diff --git a/docker/main/Dockerfile.nlp b/docker/main/Dockerfile.nlp
@@ -4,5 +4,4 @@ FROM pytorchignite/base:latest
 # Ignite NLP dependencies
 RUN pip install --upgrade --no-cache-dir transformers \
                                          spacy \
-                                         nltk \
-                                         torchtext
+                                         nltk
diff --git a/docker/test_image.py b/docker/test_image.py
@@ -54,7 +54,6 @@ def check_package(package_name, expected_version=None):
         check_package("cv2")
 
     if "nlp" in image_type:
-        check_package("torchtext")
         check_package("transformers")
 
     if "apex" in image_type: