Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
atalman committed Feb 26, 2025
1 parent eabecda commit 51bfb2c
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions tools/scripts/get_stable_cuda_version.py
Original file line number Diff line number Diff line change
@@ -1,14 +1,14 @@
#!/usr/bin/env python3

"""Return stable CUDA version for the current channel
"""
"""Return stable CUDA version for the current channel"""

import argparse
import json
import os
import sys
from typing import List


def main(args: List[str]) -> None:
import generate_binary_build_matrix

Expand All @@ -17,12 +17,12 @@ def main(args: List[str]) -> None:
"--channel",
help="Channel to use, default nightly",
type=str,
choices=["nightly", "test", "release", "all"],
choices=["nightly", "test", "release"],
default=os.getenv("CHANNEL", "nightly"),
)
options = parser.parse_args(args)
print(generate_binary_build_matrix.STABLE_CUDA_VERSIONS[options.channel])


if __name__ == "__main__":
main(sys.argv[1:])

0 comments on commit 51bfb2c

Please sign in to comment.