File tree Expand file tree Collapse file tree 1 file changed +12
-0
lines changed
tests/unit/sagemaker/modules/train Expand file tree Collapse file tree 1 file changed +12
-0
lines changed Original file line number Diff line number Diff 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)
218230def test_model_trainer_param_validation (test_case , modules_session ):
You can’t perform that action at this time.
0 commit comments