Skip to content

Commit

Permalink
Fixing sagemaker test
Browse files Browse the repository at this point in the history
  • Loading branch information
ahsimb committed Jan 31, 2024
1 parent 8dbe026 commit aba94a7
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions test/notebooks/nbtest_sagemaker.py
Original file line number Diff line number Diff line change
Expand Up @@ -190,9 +190,8 @@ def test_sagemaker(access_to_temp_secret_store, uploading_hack):
os.chdir('../sagemaker')
run_notebook('sme_init.ipynb', store_file, store_password,
hacks=[uploading_hack])
run_notebook(
'../../exasol/ds/sandbox/runtime/ansible/roles/jupyter/files/notebook/sagemaker/sme_train_model.ipynb', store_file, store_password,
hacks=[get_job_polling_hack()])
run_notebook('sme_train_model.ipynb', store_file, store_password,
hacks=[get_job_polling_hack()])
run_notebook('sme_deploy_model.ipynb', store_file, store_password)
finally:
os.chdir(current_dir)
Expand Down

0 comments on commit aba94a7

Please sign in to comment.