From 2864bb3d51dcb1c40b915e365a18f1aefe08579e Mon Sep 17 00:00:00 2001
From: Omkar Salpekar <osalpekar@gmail.com>
Date: Tue, 5 Sep 2023 13:47:28 -0700
Subject: [PATCH] Torchmetrics in S3 Index

We will need the stable torchmetrics wheel in the S3 index, since torchrec depends on it. This is similar to how pytorch depends on numpy, etc. and these binaries need to be hosted in our index when uses try to pip install from download.pytorch.org.
---
 s3_management/manage.py | 1 +
 1 file changed, 1 insertion(+)

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",