Skip to content

Commit b095a46

Browse files
committed
fix: jumpstart docs
1 parent 7aa181d commit b095a46

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

doc/overview.rst

+2-3
Original file line numberDiff line numberDiff line change
@@ -741,6 +741,7 @@ see `Model <https://sagemaker.readthedocs.io/en/stable/api/inference/model.html
741741
.. code:: python
742742
743743
from sagemaker.model import Model
744+
from sagemaker.predictor import Predictor
744745
from sagemaker.session import Session
745746
746747
# Create the SageMaker model instance
@@ -750,6 +751,7 @@ see `Model <https://sagemaker.readthedocs.io/en/stable/api/inference/model.html
750751
   source_dir=script_uri,
751752
   entry_point="inference.py",
752753
   role=Session().get_caller_identity_arn(),
754+
   predictor_cls=Predictor,
753755
)
754756
755757
Save the output from deploying the model to a variable named
@@ -761,12 +763,9 @@ Deployment may take about 5 minutes.
761763

762764
.. code:: python
763765
764-
from sagemaker.predictor import Predictor
765-
766766
predictor = model.deploy(
767767
   initial_instance_count=instance_count,
768768
   instance_type=instance_type,
769-
   predictor_cls=Predictor
770769
)
771770
772771
Because ``catboost`` relies on the PyTorch Deep Learning Containers

0 commit comments

Comments
 (0)