[Docker] Add commit as label (#20504)

Adds the Ray commit sha as a label for the docker image.
This commit is contained in:
Amog Kamsetty 2021-11-17 15:20:41 -08:00 committed by GitHub
parent 1cadd61917
commit 4cbcb11458
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -211,7 +211,8 @@ def _build_docker_image(image_name: str,
labels = {
"image-name": image_name,
"python-version": PY_MATRIX[py_version]
"python-version": PY_MATRIX[py_version],
"ray-commit": _get_commit_sha()
}
if image_type in CUDA_FULL:
labels["cuda-version"] = CUDA_FULL[image_type]