doc: minor updates to doc strings (#520)

Co-authored-by: Aaron Markham <[email protected]>

Author: 2 people

src/sagemaker/fw_utils.py

@@ -77,14 +77,14 @@ def validate_source_dir(script, directory):
 
 
 def get_mp_parameters(distribution):
-    """Get the model parallelism parameters provided by the user
+    """Get the model parallelism parameters provided by the user.
 
     Args:
-        distribution: distribution dictionary defined by the user
+        distribution: distribution dictionary defined by the user.
 
     Returns:
         params: dictionary containing model parallelism parameters
-        to be used for training
+        used for training.
     """
     try:
         mp_dict = distribution["smdistributed"]["modelparallel"]

src/sagemaker/pytorch/estimator.py

@@ -103,9 +103,9 @@ def __init__(
                 ``image_uri`` is required. If also ``None``, then a ``ValueError``
                 will be raised.
             distribution (dict): A dictionary with information on how to run distributed training
-                (default: None). Currently we support distributed training with parameter servers,
-                Model Parallelism, Data Parallelism, and MPI. Model Parallelism can only be used
-                with MPI.
+                (default: None).  Currently, the following are supported:
+                distributed training with parameter servers, SageMaker Distributed (SMD) Data
+                and Model Parallelism, and MPI. SMD Model Parallelism can only be used with MPI.
                 To enable parameter server use the following setup:
 
                 .. code:: python

src/sagemaker/tensorflow/estimator.py

@@ -81,9 +81,9 @@ def __init__(
                 ``image_uri`` is required. If also ``None``, then a ``ValueError``
                 will be raised.
             distribution (dict): A dictionary with information on how to run distributed training
-                (default: None). Currently we support distributed training with parameter servers,
-                Model Parallelism, Data Parallelism, and MPI. Model Parallelism can only be used
-                with MPI.
+                (default: None). Currently, the following are supported:
+                distributed training with parameter servers, SageMaker Distributed (SMD) Data
+                and Model Parallelism, and MPI. SMD Model Parallelism can only be used with MPI.
                 To enable parameter server use the following setup:
 
                 .. code:: python