Skip to content

Commit 071d6cb

Browse files
committed
add unit test
1 parent 8571a99 commit 071d6cb

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

tests/unit/sagemaker/modules/train/test_model_trainer.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -202,6 +202,17 @@ def model_trainer():
202202
},
203203
"should_throw": False,
204204
},
205+
{
206+
"init_params": {
207+
"training_image": DEFAULT_IMAGE,
208+
"source_code": SourceCode(
209+
source_dir=DEFAULT_SOURCE_DIR,
210+
command="python custom_script.py",
211+
ignore_patterns=["data"]
212+
),
213+
},
214+
"should_throw": False,
215+
},
205216
],
206217
ids=[
207218
"no_params",
@@ -213,6 +224,7 @@ def model_trainer():
213224
"supported_source_code_local_tar_file",
214225
"supported_source_code_s3_dir",
215226
"supported_source_code_s3_tar_file",
227+
"supported_source_code_ignore_patterns"
216228
],
217229
)
218230
def test_model_trainer_param_validation(test_case, modules_session):

0 commit comments

Comments
 (0)