@@ -18,8 +18,6 @@ ${RAY_TORCH_CUDA_IMAGE_3.11} quay.io/rhoai/ray@sha256:5077f9bb230dfa
1818${RAY_ROCM_IMAGE_3.11 } quay.io/modh/ray@sha256:48c7864989c8f22ef07772c698b761f5c1b58c6781e7a99518204d521bf56a4c
1919# Corresponds to quay.io/rhoai/ray:2.35.0-py311-rocm61-torch24-fa26
2020${RAY_TORCH_ROCM_IMAGE_3.11 } quay.io/rhoai/ray@sha256:b0e129cd2f4cdea7ad7a7859031357ffd9915410551f94fbcb942af2198cdf78
21- # Corresponds to quay.io/modh/fms-hf-tuning:v2.8.2
22- ${FMS_HF_TUNING_IMAGE } quay.io/modh/fms-hf-tuning@sha256:1ad46fe1a23f41f190c49ec2549c64f484c88fe220888a7a5700dd857ca243cc
2321# Corresponds to quay.io/modh/training:py311-cuda121-torch241
2422${CUDA_TRAINING_IMAGE_TORCH241 } quay.io/modh/training@sha256:f64f7bba3f1020d39491ac84d40d362a52e4822bdc11a33cfff021178b7c4097
2523# Corresponds to quay.io/modh/training:py311-rocm62-torch241
@@ -32,7 +30,6 @@ ${ROCM_TRAINING_IMAGE_TORCH251} quay.io/modh/training@sha256:6cdae840fa
3230${NOTEBOOK_IMAGE_3.11 } quay.io/modh/odh-workbench-jupyter-datascience-cpu-py311-ubi9@sha256:48c2be818e8d2a5005a69e2c76f9a2d40ddeb1d03376e04516ca6da13418c887
3331${NOTEBOOK_USER_NAME } ${TEST_USER_3.USERNAME }
3432${NOTEBOOK_USER_PASSWORD } ${TEST_USER_3.PASSWORD }
35- ${FMS_BINARY_NAME } fms
3633${KFTO_BINARY_NAME } kfto
3734${ODH_BINARY_NAME } odh
3835${PIP_INDEX_URL } ${PIP_INDEX_URL }
@@ -109,10 +106,6 @@ Prepare Training Operator KFTO E2E Test Suite
109106 [Documentation] Prepare Training Operator KFTO E2E Test Suite
110107 Prepare Training Operator E2E Test Suite ${KFTO_BINARY_NAME }
111108
112- Prepare Training Operator FMS E2E Test Suite
113- [Documentation] Prepare Training Operator FMS E2E Test Suite
114- Prepare Training Operator E2E Test Suite ${FMS_BINARY_NAME }
115-
116109Prepare Training Operator E2E Test Suite
117110 [Documentation] Prepare Training Operator E2E Test Suite
118111 [Arguments] ${test_binary }
@@ -130,39 +123,12 @@ Teardown Training Operator KFTO E2E Test Suite
130123 [Documentation] Teardown Training Operator KFTO E2E Test Suite
131124 Teardown Training Operator E2E Test Suite ${KFTO_BINARY_NAME }
132125
133- Teardown Training Operator FMS E2E Test Suite
134- [Documentation] Teardown Training Operator FMS E2E Test Suite
135- Teardown Training Operator E2E Test Suite ${FMS_BINARY_NAME }
136-
137126Teardown Training Operator E2E Test Suite
138127 [Documentation] Teardown Training Operator E2E Test Suite
139128 [Arguments] ${test_binary }
140129 Log To Console "Removing test binaries"
141130 Remove File ${test_binary }
142131
143- Run Training Operator FMS Test
144- [Documentation] Run Training Operator FMS Test
145- [Arguments] ${TEST_NAME }
146- Log To Console "Running test: ${TEST_NAME } "
147- ${result } = Run Process ./${FMS_BINARY_NAME } -test.run ${TEST_NAME }
148- ... shell=true
149- ... stderr=STDOUT
150- ... env:TEST_TIMEOUT_SHORT=5m
151- ... env:TEST_TIMEOUT_MEDIUM=10m
152- ... env:TEST_TIMEOUT_LONG=20m
153- ... env:TEST_OUTPUT_DIR=%{WORKSPACE } /codeflare-${FMS_BINARY_NAME } -logs
154- ... env:FMS_HF_TUNING_IMAGE=${FMS_HF_TUNING_IMAGE }
155- ... env:AWS_DEFAULT_ENDPOINT=${AWS_DEFAULT_ENDPOINT }
156- ... env:AWS_ACCESS_KEY_ID=${AWS_ACCESS_KEY_ID }
157- ... env:AWS_SECRET_ACCESS_KEY=${AWS_SECRET_ACCESS_KEY }
158- ... env:AWS_STORAGE_BUCKET_DOWNLOAD=${AWS_STORAGE_BUCKET }
159- ... env:AWS_STORAGE_BUCKET_DOWNLOAD_MODEL_PATH=models/granite-3b-code-base-2k
160- Log To Console ${result.stdout }
161- Check missing Go test ${result.stdout }
162- IF ${result.rc } != 0
163- FAIL ${TEST_NAME } failed
164- END
165-
166132Run Training Operator KFTO Test
167133 [Documentation] Run Training Operator KFTO Test
168134 [Arguments] ${TEST_NAME }
0 commit comments