fix docstring for decorated functions

Author: jerrypeng7773

doc/requirements.txt

@@ -1,4 +1,4 @@
-sphinx==3.1.1
+sphinx==3.1.2
 sphinx-rtd-theme==0.5.0
 docutils==0.15.2
 packaging==20.9

src/sagemaker/workflow/pipeline_context.py

@@ -16,6 +16,7 @@
 import warnings
 import inspect
 from typing import Dict
+from functools import wraps
 
 from sagemaker.session import Session, SessionSettings
 
@@ -131,7 +132,7 @@ def runnable_by_pipeline(run_func):
     the arguments needed to compose that particular step as part of the pipeline.
     The job will be started during pipeline execution.
     """
-
+    @wraps(run_func)
     def wrapper(*args, **kwargs):
         self_instance = args[0]
         if isinstance(self_instance.sagemaker_session, PipelineSession):