From 4168982dadb79fc7b4ada81de69123d5f6830d76 Mon Sep 17 00:00:00 2001 From: Aaron Orenstein Date: Mon, 3 Feb 2025 09:27:27 -0800 Subject: [PATCH] PEP585: .github release triggers (#145708) Pull Request resolved: https://github.com/pytorch/pytorch/pull/145708 Approved by: https://github.com/malfet --- .github/scripts/generate_docker_release_matrix.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/.github/scripts/generate_docker_release_matrix.py b/.github/scripts/generate_docker_release_matrix.py index 6bccffb6069..0f9f3ef3021 100644 --- a/.github/scripts/generate_docker_release_matrix.py +++ b/.github/scripts/generate_docker_release_matrix.py @@ -12,7 +12,6 @@ Will output a condensed version of the matrix. Will include fllowing: """ import json -from typing import Dict, List import generate_binary_build_matrix @@ -20,8 +19,8 @@ import generate_binary_build_matrix DOCKER_IMAGE_TYPES = ["runtime", "devel"] -def generate_docker_matrix() -> Dict[str, List[Dict[str, str]]]: - ret: List[Dict[str, str]] = [] +def generate_docker_matrix() -> dict[str, list[dict[str, str]]]: + ret: list[dict[str, str]] = [] # CUDA amd64 Docker images are available as both runtime and devel while # CPU arm64 image is only available as runtime. for cuda, version in generate_binary_build_matrix.CUDA_ARCHES_FULL_VERSION.items():