Skip to content

Commit

Permalink
test
Browse files Browse the repository at this point in the history
  • Loading branch information
lanluo-nvidia committed Oct 28, 2024
1 parent 3206da7 commit 0a64986
Showing 1 changed file with 14 additions and 3 deletions.
17 changes: 14 additions & 3 deletions .github/scripts/generate-tensorrt-test-matrix.py
Original file line number Diff line number Diff line change
Expand Up @@ -61,10 +61,21 @@ def main(args: list[str]) -> None:
matrix_dict = json.loads(options.matrix)
includes = matrix_dict["include"]
assert len(includes) > 0
channel = includes[0].channel
if "windows" in includes[0].validation_runner:
if "channel" not in includes[0]:
raise Exception(f"channel field is missing from the matrix: {options.matrix}")
channel = includes[0]["channel"]
if channel not in ("nightly", "test", "release"):
raise Exception(
f"channel field: {channel} is not supported, currently supported value: nightly, test, release"
)

if "validation_runner" not in includes[0]:
raise Exception(
f"validation_runner field is missing from the matrix: {options.matrix}"
)
if "windows" in includes[0]["validation_runner"]:
arch = "windows"
elif "linux" in includes[0].validation_runner:
elif "linux" in includes[0]["validation_runner"]:
arch = "linux"
else:
raise Exception(
Expand Down

0 comments on commit 0a64986

Please sign in to comment.