File tree 1 file changed +2
-3
lines changed
1 file changed +2
-3
lines changed Original file line number Diff line number Diff line change 38
38
from sagemaker .workflow .entities import Expression
39
39
from sagemaker .dataset_definition .inputs import S3Input , DatasetDefinition
40
40
from sagemaker .apiutils ._base_types import ApiObject
41
- from sagemaker .estimator import Framework
42
41
from sagemaker .s3 import S3Uploader
43
42
44
43
@@ -1256,7 +1255,7 @@ def __init__(
1256
1255
to be run as part of the Processing Job.
1257
1256
1258
1257
Args:
1259
- estimator_cls (type): A subclass of the :class:`~sagemaker.network.NetworkConfig `
1258
+ estimator_cls (type): A subclass of the :class:`~sagemaker.estimator.Framework `
1260
1259
estimator
1261
1260
framework_version (str): The version of the framework
1262
1261
s3_prefix (str): The S3 prefix URI where custom code will be
@@ -1496,7 +1495,7 @@ def _upload_payload(
1496
1495
dependencies : Optional [List [str ]],
1497
1496
git_config : Optional [Dict [str , str ]],
1498
1497
job_name : str ,
1499
- ) -> Framework :
1498
+ ) -> "sagemaker.estimator. Framework" :
1500
1499
# A new estimator instance is required, because each call to ScriptProcessor.run() can
1501
1500
# use different codes.
1502
1501
estimator = self .estimator_cls (
You can’t perform that action at this time.
0 commit comments