Skip to content

Commit 547e1e6

Browse files
committed
Fix circular dependency between processing.py and estimator.py
1 parent b83b0d0 commit 547e1e6

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

src/sagemaker/processing.py

+2-3
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,6 @@
3838
from sagemaker.workflow.entities import Expression
3939
from sagemaker.dataset_definition.inputs import S3Input, DatasetDefinition
4040
from sagemaker.apiutils._base_types import ApiObject
41-
from sagemaker.estimator import Framework
4241
from sagemaker.s3 import S3Uploader
4342

4443

@@ -1256,7 +1255,7 @@ def __init__(
12561255
to be run as part of the Processing Job.
12571256
12581257
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`
12601259
estimator
12611260
framework_version (str): The version of the framework
12621261
s3_prefix (str): The S3 prefix URI where custom code will be
@@ -1496,7 +1495,7 @@ def _upload_payload(
14961495
dependencies: Optional[List[str]],
14971496
git_config: Optional[Dict[str, str]],
14981497
job_name: str,
1499-
) -> Framework:
1498+
) -> "sagemaker.estimator.Framework":
15001499
# A new estimator instance is required, because each call to ScriptProcessor.run() can
15011500
# use different codes.
15021501
estimator = self.estimator_cls(

0 commit comments

Comments
 (0)