Add missing type hint

Author: Payton Staub

src/sagemaker/workflow/steps.py

@@ -25,6 +25,7 @@
     CreateModelInput,
     TrainingInput,
     TransformInput,
+    FileSystemInput
 )
 from sagemaker.model import Model
 from sagemaker.processing import (
@@ -44,7 +45,6 @@
     Properties,
 )
 
-
 class StepTypeEnum(Enum, metaclass=DefaultEnumMeta):
     """Enum of step types."""
 
@@ -145,7 +145,7 @@ def __init__(
         self,
         name: str,
         estimator: EstimatorBase,
-        inputs: Union[TrainingInput, dict, str] = None,
+        inputs: Union[TrainingInput, dict, str, FileSystemInput] = None,
         cache_config: CacheConfig = None,
         depends_on: List[str] = None,
     ):
@@ -157,7 +157,8 @@ def __init__(
         Args:
             name (str): The name of the training step.
             estimator (EstimatorBase): A `sagemaker.estimator.EstimatorBase` instance.
-            inputs (str or dict or sagemaker.inputs.TrainingInput): Information
+            inputs (str or dict or sagemaker.inputs.TrainingInput
+                or sagemaker.inputs.FileSystemInput): Information
                 about the training data. This can be one of three types:
 
                 * (str) the S3 location where training data is saved, or a file:// path in
@@ -169,7 +170,7 @@ def __init__(
                     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.