Skip to content

Commit f7f1160

Browse files
committed
Merge remote-tracking branch 'upstream/main' into rhoai-3.2
2 parents d7d9f06 + 7e3f68d commit f7f1160

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

tests/trainer/sdk_tests/fashion_mnist_tests.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ func RunFashionMnistCpuDistributedTraining(t *testing.T) {
8787
"export AWS_DEFAULT_ENDPOINT='%s'; export AWS_ACCESS_KEY_ID='%s'; "+
8888
"export AWS_SECRET_ACCESS_KEY='%s'; export AWS_STORAGE_BUCKET='%s'; "+
8989
"export AWS_STORAGE_BUCKET_MNIST_DIR='%s'; "+
90-
"python -m pip install --quiet --no-cache-dir papermill boto3==1.34.162 && "+
90+
"python -m pip install --quiet --no-cache-dir ipykernel papermill boto3==1.34.162 && "+
9191
"if python -m papermill -k python3 /opt/app-root/notebooks/%s /opt/app-root/src/out.ipynb --log-output; "+
9292
"then echo 'NOTEBOOK_STATUS: SUCCESS'; else echo 'NOTEBOOK_STATUS: FAILURE'; fi; sleep infinity",
9393
support.GetOpenShiftApiUrl(test), userToken, namespace.Name, rwxPvc.Name,

tests/trainer/sdk_tests/osft_traininghub_tests.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,7 @@ func RunOsftTrainingHubMultiGpuDistributedTraining(t *testing.T) {
8989
"export AWS_SECRET_ACCESS_KEY='%s'; "+
9090
"export AWS_STORAGE_BUCKET='%s'; "+
9191
"export AWS_STORAGE_BUCKET_OSFT_DIR='%s'; "+
92-
"python -m pip install --quiet --no-cache-dir --break-system-packages papermill boto3==1.34.162 && "+
92+
"python -m pip install --quiet --no-cache-dir --break-system-packages ipykernel papermill boto3==1.34.162 && "+
9393
"if python -m papermill -k python3 /opt/app-root/notebooks/%s /opt/app-root/src/out.ipynb --log-output; "+
9494
"then echo 'NOTEBOOK_STATUS: SUCCESS'; else echo 'NOTEBOOK_STATUS: FAILURE'; fi; sleep infinity",
9595
support.GetOpenShiftApiUrl(test), userToken, namespace.Name, rwxPvc.Name,

tests/trainer/sdk_tests/sft_traininghub_tests.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -89,7 +89,7 @@ func RunSftTrainingHubMultiGpuDistributedTraining(t *testing.T) {
8989
"export AWS_SECRET_ACCESS_KEY='%s'; "+
9090
"export AWS_STORAGE_BUCKET='%s'; "+
9191
"export AWS_STORAGE_BUCKET_SFT_DIR='%s'; "+
92-
"python -m pip install --quiet --no-cache-dir --break-system-packages papermill boto3==1.34.162 && "+
92+
"python -m pip install --quiet --no-cache-dir --break-system-packages ipykernel papermill boto3==1.34.162 && "+
9393
"if python -m papermill -k python3 /opt/app-root/notebooks/%s /opt/app-root/src/out.ipynb --log-output; "+
9494
"then echo 'NOTEBOOK_STATUS: SUCCESS'; else echo 'NOTEBOOK_STATUS: FAILURE'; fi; sleep infinity",
9595
support.GetOpenShiftApiUrl(test), userToken, namespace.Name, rwxPvc.Name,

0 commit comments

Comments
 (0)