Output empty json if version not provided to release matrix (#4742)

For release matrix generation. Allow empty version as input.
pytorch · Nov 16, 2023 · 32bcfef · 32bcfef
1 parent da280fe
commit 32bcfef
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/.github/workflows/generate_release_matrix.yml b/.github/workflows/generate_release_matrix.yml
@@ -5,7 +5,7 @@ on:
     inputs:
       version:
         description: "Release matrix for pytorch release version"
-        default: "2.1.1"
+        default: ""
         type: string
       test-infra-repository:
         description: "Test infra repository to use"

diff --git a/tools/scripts/generate_release_matrix.py b/tools/scripts/generate_release_matrix.py
@@ -27,15 +27,16 @@ def main(args) -> None:
         "--version",
         help="PyTorch Release version",
         type=str,
-        default=os.getenv("VERSION", "2.1.1"),
+        default=os.getenv("VERSION", ""),
     )
-
     options = parser.parse_args(args)
 
-    if options.version not in RELEASE_DICT.keys():
+    if options.version and options.version not in RELEASE_DICT.keys():
         raise ValueError(f"{options.version} is not a valid release")
-
-    print(json.dumps(RELEASE_DICT[options.version]))
+    elif options.version:
+        print(json.dumps(RELEASE_DICT[options.version]))
+    else:
+        print(json.dumps({}))
 
 
 if __name__ == "__main__":