fix package sync (#135)

.github/issue-templates/packages-out-of-sync.md

@@ -1,5 +1,5 @@
 ---
-title: `light-the-torch` is out of sync with the PyTorch package indices
+title: light-the-torch is out of sync with the PyTorch package indices
 assignees: pmeier
 ---
 {%- if env.MISSING|length %}

.github/workflows/check-pytorch-package-indices.yml

@@ -29,10 +29,10 @@ jobs:
           OUT=$(python scripts/check_pytorch_package_indices.py)
 
           MISSING=$(echo $OUT | jq -r '.missing | join(",")')
-          echo "missing=${MISSING}" >> $GITHUB_OUTPUT
+          echo "missing=${MISSING}" | tee -a "${GITHUB_OUTPUT}"
 
           EXTRA=$(echo $OUT | jq -r '.extra | join(",")')
-          echo "extra=${EXTRA}" >> $GITHUB_OUTPUT
+          echo "extra=${EXTRA}" | tee -a "${GITHUB_OUTPUT}"
 
           [ -z "${MISSING}${EXTRA}" ];
 

light_the_torch/_patch.py

@@ -49,18 +49,24 @@ PYTORCH_DISTRIBUTIONS = {
 }
 
 THIRD_PARTY_PACKAGES = {
+    "Jinja2",
+    "MarkupSafe",
     "Pillow",
     "certifi",
     "charset-normalizer",
     "cmake",
+    "colorama",
     "filelock",
     "idna",
+    "lit",
     "mpmath",
     "networkx",
     "numpy",
     "packaging",
+    "portalocker",
     "requests",
     "sympy",
+    "tqdm",
     "typing-extensions",
     "urllib3",
 }

pyproject.toml

@@ -66,7 +66,7 @@ local_scheme = "node-and-timestamp"
 # https://github.com/psf/black#configuration-format
 
 line-length = 88
-target-version = ['py37', 'py38', 'py39', 'py310', 'py311']
+target-version = ['py37']
 exclude = '''
 /(
     \.git

scripts/check_pytorch_package_indices.py

@@ -17,6 +17,7 @@ EXCLUDED_PYTORCH_PACKAGES = {
     "nestedtensor",
     "pytorch_csprng",
     "pytorch-triton",
+    "pytorch-triton-rocm",
     "torch-cuda80",
     "torch-nightly",
     "torchaudio_nightly",
@@ -28,6 +29,7 @@ EXCLUDED_PYTORCH_PACKAGES = {
     "torchrec_nightly_3.9_cu11.whl",
     "torchrec_nightly_cpu",
     "torchtriton",
+    "triton",
 }
 HANDLED_PACKAGES = PYTORCH_DISTRIBUTIONS | THIRD_PARTY_PACKAGES