fix docstring for decorated functions (aws#3115)

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
 
@@ -132,6 +133,7 @@ def runnable_by_pipeline(run_func):
     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):