@@ -366,6 +366,7 @@ def test_register_model_sip(estimator, model_metrics):
366
366
367
367
def test_register_model_with_model_repack_with_estimator (estimator , model_metrics ):
368
368
model_data = f"s3://{ BUCKET } /model.tar.gz"
369
+ dummy_requirements = f"{ DATA_DIR } /dummy_requirements.txt"
369
370
register_model = RegisterModel (
370
371
name = "RegisterModelStep" ,
371
372
estimator = estimator ,
@@ -379,6 +380,7 @@ def test_register_model_with_model_repack_with_estimator(estimator, model_metric
379
380
approval_status = "Approved" ,
380
381
description = "description" ,
381
382
entry_point = f"{ DATA_DIR } /dummy_script.py" ,
383
+ dependencies = [dummy_requirements ],
382
384
depends_on = ["TestStep" ],
383
385
tags = [{"Key" : "myKey" , "Value" : "myValue" }],
384
386
)
@@ -405,6 +407,7 @@ def test_register_model_with_model_repack_with_estimator(estimator, model_metric
405
407
},
406
408
"HyperParameters" : {
407
409
"inference_script" : '"dummy_script.py"' ,
410
+ "dependencies" : f"\" { dummy_requirements } \" " ,
408
411
"model_archive" : '"model.tar.gz"' ,
409
412
"sagemaker_submit_directory" : '"s3://{}/{}/source/sourcedir.tar.gz"' .format (
410
413
BUCKET , repacker_job_name .replace ('"' , "" )
@@ -413,6 +416,7 @@ def test_register_model_with_model_repack_with_estimator(estimator, model_metric
413
416
"sagemaker_container_log_level" : "20" ,
414
417
"sagemaker_job_name" : repacker_job_name ,
415
418
"sagemaker_region" : f'"{ REGION } "' ,
419
+ "source_dir" : 'null'
416
420
},
417
421
"InputDataConfig" : [
418
422
{
@@ -631,6 +635,7 @@ def test_register_model_with_model_repack_with_pipeline_model(pipeline_model, mo
631
635
"S3OutputPath" : f"s3://{ BUCKET } /" ,
632
636
},
633
637
"HyperParameters" : {
638
+ "dependencies" : 'null' ,
634
639
"inference_script" : '"dummy_script.py"' ,
635
640
"model_archive" : '"model.tar.gz"' ,
636
641
"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
640
645
"sagemaker_container_log_level" : "20" ,
641
646
"sagemaker_job_name" : repacker_job_name ,
642
647
"sagemaker_region" : f'"{ REGION } "' ,
648
+ "source_dir" : 'null' ,
643
649
},
644
650
"InputDataConfig" : [
645
651
{
0 commit comments