fix: linting

aws · Jun 22, 2022 · 6c077b3 · 6c077b3
1 parent d4d4b83
commit 6c077b3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/unit/sagemaker/training_compiler/test_tensorflow_compiler.py b/tests/unit/sagemaker/training_compiler/test_tensorflow_compiler.py
@@ -226,7 +226,7 @@ def test_mwms(
                 framework_version=tensorflow_training_version,
                 enable_sagemaker_metrics=False,
                 compiler_config=TrainingCompilerConfig(),
-                distribution={"multi_worker_mirrored_strategy": True},
+                distribution={"multi_worker_mirrored_strategy": {"enabled": True}},
             ).fit()
 
     def test_python_2(