fix: linting

Author: Lokiiiiii

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(