ray-train: make all Dockerfile ARGs config-driven

Every Dockerfile ARG must come from the config; the Dockerfile default is only a
fallback for bare 'docker build'. Add the missing ones to both configs:
- builder_base_image, base_image (were relying on Dockerfile defaults)
- python_short_version (venv site-packages path; was Dockerfile-only)
Drop redundant build.framework_version — the build-image action already injects
FRAMEWORK_VERSION from metadata.framework_version (kept in metadata as required).

Author: Eren-Jeager123

.github/config/image/ray-train/ec2-gpu.yml

@@ -16,9 +16,11 @@ metadata:
 build:
   dockerfile: "docker/ray-train/Dockerfile.cuda"
   target: "ec2"
+  builder_base_image: "public.ecr.aws/deep-learning-containers/base:devel-cu130-amzn2023"
+  base_image: "public.ecr.aws/deep-learning-containers/base:runtime-cu130-amzn2023"
   python_version: "3.13.12"
+  python_short_version: "3.13"
   cuda_version: "13.0.2"
-  framework_version: "2.56.0"
   torch_version: "2.13.0"
   efa_version: "1.47.0"
   gdrcopy_version: "2.4.4"

.github/config/image/ray-train/eks-gpu.yml

@@ -17,9 +17,11 @@ metadata:
 build:
   dockerfile: "docker/ray-train/Dockerfile.cuda"
   target: "eks"
+  builder_base_image: "public.ecr.aws/deep-learning-containers/base:devel-cu130-amzn2023"
+  base_image: "public.ecr.aws/deep-learning-containers/base:runtime-cu130-amzn2023"
   python_version: "3.13.12"
+  python_short_version: "3.13"
   cuda_version: "13.0.2"
-  framework_version: "2.56.0"
   torch_version: "2.13.0"
   efa_version: "1.47.0"
   gdrcopy_version: "2.4.4"