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 @@ -741,6 +741,7 @@ see `Model <https://sagemaker.readthedocs.io/en/stable/api/inference/model.html
741
741
.. code :: python
742
742
743
743
from sagemaker.model import Model
744
+ from sagemaker.predictor import Predictor
744
745
from sagemaker.session import Session
745
746
746
747
# Create the SageMaker model instance
@@ -750,6 +751,7 @@ see `Model <https://sagemaker.readthedocs.io/en/stable/api/inference/model.html
750
751
source_dir = script_uri,
751
752
entry_point = " inference.py" ,
752
753
role = Session().get_caller_identity_arn(),
754
+ predictor_cls = Predictor,
753
755
)
754
756
755
757
Save the output from deploying the model to a variable named
@@ -761,12 +763,9 @@ Deployment may take about 5 minutes.
761
763
762
764
.. code :: python
763
765
764
- from sagemaker.predictor import Predictor
765
-
766
766
predictor = model.deploy(
767
767
initial_instance_count = instance_count,
768
768
instance_type = instance_type,
769
- predictor_cls = Predictor
770
769
)
771
770
772
771
Because ``catboost `` relies on the PyTorch Deep Learning Containers
You can’t perform that action at this time.
0 commit comments