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():
202
202
},
203
203
"should_throw" : False ,
204
204
},
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
+ },
205
216
],
206
217
ids = [
207
218
"no_params" ,
@@ -213,6 +224,7 @@ def model_trainer():
213
224
"supported_source_code_local_tar_file" ,
214
225
"supported_source_code_s3_dir" ,
215
226
"supported_source_code_s3_tar_file" ,
227
+ "supported_source_code_ignore_patterns"
216
228
],
217
229
)
218
230
def test_model_trainer_param_validation (test_case , modules_session ):
You can’t perform that action at this time.
0 commit comments