diff --git a/s3_management/manage.py b/s3_management/manage.py index 655f7de40..4b35ecab2 100644 --- a/s3_management/manage.py +++ b/s3_management/manage.py @@ -85,6 +85,7 @@ "torchcsprng", "torchdata", "torchdistx", + "torchmetrics", "torchrec", "torchtext", "torchvision",