Skip to content

Commit c24c5f5

Browse files
committed
Configuring tests to use the new hyperparameter for MWMS
1 parent 36eb70d commit c24c5f5

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

test/integration/sagemaker/test_multi_worker_mirrored.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,10 @@ def test_multi_node(sagemaker_session, instance_type, image_uri, tmpdir, framewo
3737
image_name=image_uri,
3838
framework_version=framework_version,
3939
py_version="py3",
40+
hyperparameters={
41+
'sagemaker_multi_worker_mirrored_enabled': True,
42+
},
4043
sagemaker_session=sagemaker_session,
4144
)
4245
estimator.fit(job_name=unique_name_from_base("test-tf-mwms"))
43-
raise NotImplementedError('Yet to add assertions')
46+
raise NotImplementedError('Yet to add assertion')

0 commit comments

Comments
 (0)