fix: changing variable name for MWMS

aws · Jun 3, 2022 · 994a321 · 994a321
1 parent 64b7dec
commit 994a321
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/sagemaker_training/params.py b/src/sagemaker_training/params.py
@@ -35,7 +35,7 @@
 DEFAULT_MODULE_NAME_PARAM = "default_user_module_name"  # type: str
 MPI_ENABLED = "sagemaker_mpi_enabled"  # type: str
 PARAMETER_SERVER_ENABLED = "sagemaker_parameter_server_enabled"  # type: str
-MULTI_WORKER_MIRRORED_ENABLED = "sagemaker_multi_worker_mirrored_enabled"  # type: str
+MULTI_WORKER_MIRRORED_ENABLED = "sagemaker_multi_worker_mirrored_strategy_enabled"  # type: str
 REGION_NAME_PARAM = "sagemaker_region"  # type: str
 REGION_NAME_ENV = REGION_NAME_PARAM.upper()  # type: str
 DEFAULT_INVOCATIONS_ACCEPT_ENV = "SAGEMAKER_DEFAULT_INVOCATIONS_ACCEPT"  # type: str