exclude torchrec from patched dists (#85)

pmeier · Jul 16, 2022 · fea8800 · fea8800
1 parent c8d9b1d
commit fea8800
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/scripts/check_available_pytorch_dists.py b/scripts/check_available_pytorch_dists.py
@@ -14,10 +14,13 @@
     "torch_cuda80",
     "torch_nightly",
     "torchaudio_nightly",
+    "torchrec",
+    "torchrec_cpu",
     "torchrec_nightly",
     "torchrec_nightly_3.7_cu11.whl",
     "torchrec_nightly_3.8_cu11.whl",
     "torchrec_nightly_3.9_cu11.whl",
+    "torchrec_nightly_cpu",
 }
 PATCHED_PYTORCH_DISTS = set(PYTORCH_DISTRIBUTIONS)