Update main.yml

Author: apoorvkh

.github/workflows/main.yml

@@ -51,7 +51,11 @@ jobs:
       - name: Get PyTorch versions
         id: get-pytorch-versions
         run: |
-          VERSIONS=($(curl -s https://pypi.org/pypi/torch/json | jq -r '.releases | keys[]' | grep -v '^1\.' | grep -E '\.[0]+$' | sort -rV | sed 's/\.0$//') | jq -R . | jq -sc .)
+          VERSIONS=$(
+            curl -s https://pypi.org/pypi/torch/json | jq -r '.releases | keys[]' |  # get PyTorch versions
+            grep -v '^1\.' | grep -E '\.[0]+$' | sort -rV | sed 's/\.0$//' |  # remove versions <2.0; strip "patch" from versions
+            jq -R . | jq -sc .  # to JSON array
+          )
           echo "versions=$VERSIONS" >> $GITHUB_OUTPUT
 
   test: