documentation: fix documentation for input types in estimator.fit (aws#2685)

* documentation: fix documentation for input types in estimator.fit

* documentation: add fourth input type to type list for inputs in doc string

* update type options for values in dict input

Co-authored-by: Ahsan Khan <[email protected]>
Co-authored-by: Jeniya Tabassum <[email protected]>

Author: 3 people

src/sagemaker/estimator.py

@@ -650,19 +650,22 @@ def fit(self, inputs=None, wait=True, logs="All", job_name=None, experiment_conf
         model using the Amazon SageMaker hosting services.
 
         Args:
-            inputs (str or dict or sagemaker.inputs.TrainingInput): Information
-                about the training data. This can be one of three types:
+            inputs (str or dict or sagemaker.inputs.TrainingInput or
+                sagemaker.inputs.FileSystemInput): Information about the training data.
+                This can be one of four types:
 
                 * (str) the S3 location where training data is saved, or a file:// path in
                     local mode.
-                * (dict[str, str] or dict[str, sagemaker.inputs.TrainingInput]) If using multiple
-                    channels for training data, you can specify a dict mapping channel names to
-                    strings or :func:`~sagemaker.inputs.TrainingInput` objects.
+                * (dict[str, str] or dict[str, sagemaker.inputs.TrainingInput] or
+                    dict[str, sagemaker.inputs.FileSystemInput]) If using multiple channels for
+                    training data, you can specify a dict mapping channel names to strings or
+                    :func:`~sagemaker.inputs.TrainingInput` objects or
+                    :func:`~sagemaker.inputs.FileSystemInput` objects.
                 * (sagemaker.inputs.TrainingInput) - channel configuration for S3 data sources
                     that can provide additional information as well as the path to the training
                     dataset.
                     See :func:`sagemaker.inputs.TrainingInput` for full details.
-                * (sagemaker.session.FileSystemInput) - channel configuration for
+                * (sagemaker.inputs.FileSystemInput) - channel configuration for
                     a file system data source that can provide additional information as well as
                     the path to the training dataset.