Skip to content

Commit 8c93f92

Browse files
fix docstring for decorated functions (#3115)
1 parent c006c7a commit 8c93f92

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

doc/requirements.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
sphinx==3.1.1
1+
sphinx==3.1.2
22
sphinx-rtd-theme==0.5.0
33
docutils==0.15.2
44
packaging==20.9

src/sagemaker/workflow/pipeline_context.py

+2
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
import warnings
1717
import inspect
1818
from typing import Dict
19+
from functools import wraps
1920

2021
from sagemaker.session import Session, SessionSettings
2122

@@ -132,6 +133,7 @@ def runnable_by_pipeline(run_func):
132133
The job will be started during pipeline execution.
133134
"""
134135

136+
@wraps(run_func)
135137
def wrapper(*args, **kwargs):
136138
self_instance = args[0]
137139
if isinstance(self_instance.sagemaker_session, PipelineSession):

0 commit comments

Comments
 (0)