Skip to content

Commit

Permalink
Correct extras
Browse files Browse the repository at this point in the history
Signed-off-by: Simon Zhao <simonyansenzhao@gmail.com>
  • Loading branch information
SimonYansenZhao committed Sep 26, 2024
1 parent 35a0b21 commit 2be5324
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/ci/azureml_tests/aml_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -81,7 +81,7 @@ def get_or_create_environment(
python_version (str): python version, such as "3.11"
"""
compute = "gpu" if use_gpu else "cpu"
extras = "dev" + ",gpu" if use_gpu else "" + ",spark" if use_spark else ""
extras = "dev" + (",gpu" if use_gpu else "") + (",spark" if use_spark else "")
dockerfile = pathlib.Path("tools/docker/Dockerfile")

# Docker's --build-args is not supported by AzureML Python SDK v2 as shown
Expand Down

0 comments on commit 2be5324

Please sign in to comment.