Facing an Error while trying to Import ModelStep

Question

I am trying to import ModelStep using the code

from sagemaker.workflow.model_step import ModelStep

But it is throwing the error ModuleNotFoundError: No module named 'sagemaker.workflow.model_step'

How can I resolve this issue?

Answer 1

This feature, in sagemaker-python-sdk, was introduced starting with release 2.90.0 (16 May 2022).

I suggest always keeping sagemaker updated, they come out with updates very frequently and quite important:

pip install -U sagemaker