diff --git a/s3_management/manage.py b/s3_management/manage.py
index 1ba29df45..15b37cf3d 100644
--- a/s3_management/manage.py
+++ b/s3_management/manage.py
@@ -50,6 +50,7 @@
     "networkx",
     "numpy",
     "packaging",
+    "portalocker",
     "pytorch_triton",
     "pytorch_triton_rocm",
     "requests",