Add more packages to allow list (#1344)

pytorch · Mar 9, 2023 · 9b5be0d · 9b5be0d
1 parent 8f2fdc7
commit 9b5be0d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/s3_management/manage.py b/s3_management/manage.py
@@ -39,6 +39,7 @@
     "certifi",
     "charset_normalizer",
     "cmake",
+    "colorama",
     "filelock",
     "idna",
     "Jinja2",
@@ -62,6 +63,7 @@
     "torchrec",
     "torchtext",
     "torchvision",
+    "triton",
     "tqdm",
     "typing_extensions",
     "urllib3",