Skip to content

Commit ecbc5e5

Browse files
author
Payton Staub
committed
Fix unit tests
1 parent db68957 commit ecbc5e5

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

tests/unit/sagemaker/workflow/test_step_collections.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -366,6 +366,7 @@ def test_register_model_sip(estimator, model_metrics):
366366

367367
def test_register_model_with_model_repack_with_estimator(estimator, model_metrics):
368368
model_data = f"s3://{BUCKET}/model.tar.gz"
369+
dummy_requirements = f"{DATA_DIR}/dummy_requirements.txt"
369370
register_model = RegisterModel(
370371
name="RegisterModelStep",
371372
estimator=estimator,
@@ -379,6 +380,7 @@ def test_register_model_with_model_repack_with_estimator(estimator, model_metric
379380
approval_status="Approved",
380381
description="description",
381382
entry_point=f"{DATA_DIR}/dummy_script.py",
383+
dependencies=[dummy_requirements],
382384
depends_on=["TestStep"],
383385
tags=[{"Key": "myKey", "Value": "myValue"}],
384386
)
@@ -405,6 +407,7 @@ def test_register_model_with_model_repack_with_estimator(estimator, model_metric
405407
},
406408
"HyperParameters": {
407409
"inference_script": '"dummy_script.py"',
410+
"dependencies": f"\"{dummy_requirements}\"",
408411
"model_archive": '"model.tar.gz"',
409412
"sagemaker_submit_directory": '"s3://{}/{}/source/sourcedir.tar.gz"'.format(
410413
BUCKET, repacker_job_name.replace('"', "")
@@ -413,6 +416,7 @@ def test_register_model_with_model_repack_with_estimator(estimator, model_metric
413416
"sagemaker_container_log_level": "20",
414417
"sagemaker_job_name": repacker_job_name,
415418
"sagemaker_region": f'"{REGION}"',
419+
"source_dir": 'null'
416420
},
417421
"InputDataConfig": [
418422
{
@@ -631,6 +635,7 @@ def test_register_model_with_model_repack_with_pipeline_model(pipeline_model, mo
631635
"S3OutputPath": f"s3://{BUCKET}/",
632636
},
633637
"HyperParameters": {
638+
"dependencies": 'null',
634639
"inference_script": '"dummy_script.py"',
635640
"model_archive": '"model.tar.gz"',
636641
"sagemaker_submit_directory": '"s3://{}/{}/source/sourcedir.tar.gz"'.format(
@@ -640,6 +645,7 @@ def test_register_model_with_model_repack_with_pipeline_model(pipeline_model, mo
640645
"sagemaker_container_log_level": "20",
641646
"sagemaker_job_name": repacker_job_name,
642647
"sagemaker_region": f'"{REGION}"',
648+
"source_dir": 'null',
643649
},
644650
"InputDataConfig": [
645651
{

0 commit comments

Comments
 (0)