CARRY: Use correct env variable names for test Dockerfile

Author: sutaakar

images/tests/Dockerfile

@@ -23,8 +23,8 @@ ENTRYPOINT [ "gotestsum"]
 
 # Configure images using environment variables
 ENV FMS_HF_TUNING_IMAGE=<replace_me>
-ENV CUDA_TRAINING_IMAGE_TORCH241=<replace_me>
-ENV ROCM_TRAINING_IMAGE_TORCH241=<replace_me>
-ENV CUDA_TRAINING_IMAGE_TORCH251=<replace_me>
-ENV ROCM_TRAINING_IMAGE_TORCH251=<replace_me>
+ENV TEST_TRAINING_CUDA_PYTORCH_241_IMAGE=<replace_me>
+ENV TEST_TRAINING_ROCM_PYTORCH_241_IMAGE=<replace_me>
+ENV TEST_TRAINING_CUDA_PYTORCH_251_IMAGE=<replace_me>
+ENV TEST_TRAINING_ROCM_PYTORCH_251_IMAGE=<replace_me>
 ENV NOTEBOOK_IMAGE=<replace_me>