Skip to content

Commit 9abb2de

Browse files
ajaykarpurknakad
authored andcommitted
Import Processors in init files (aws#230)
This change adds import statements in the __init__ files of the `sagemaker` package and the `sklearn` package to make it easier for users to import Processor classes.
1 parent bcf5766 commit 9abb2de

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

src/sagemaker/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,7 @@
5050
from sagemaker.model import Model, ModelPackage # noqa: F401
5151
from sagemaker.pipeline import PipelineModel # noqa: F401
5252
from sagemaker.predictor import RealTimePredictor # noqa: F401
53+
from sagemaker.processor import Processor, ScriptModeProcessor # noqa: F401
5354
from sagemaker.session import Session # noqa: F401
5455
from sagemaker.session import container_def, pipeline_container_def # noqa: F401
5556
from sagemaker.session import production_variant # noqa: F401

src/sagemaker/sklearn/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,3 +15,4 @@
1515

1616
from sagemaker.sklearn.estimator import SKLearn # noqa: F401
1717
from sagemaker.sklearn.model import SKLearnModel, SKLearnPredictor # noqa: F401
18+
from sagemaker.sklearn.processor import SKLearnProcessor # noqa: F401

0 commit comments

Comments
 (0)