From 74f240c36e18be23a5d09c8f469d720e2d246307 Mon Sep 17 00:00:00 2001
From: atalman <atalman@fb.com>
Date: Thu, 9 Mar 2023 12:00:42 -0800
Subject: [PATCH] Add more packages to allow list

---
 s3_management/manage.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/s3_management/manage.py b/s3_management/manage.py
index c6d7d535a..1ba29df45 100644
--- 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",