ImportError: Cannot import 'models' from sagemaker.jumpstart

[yishairasowsky]

Describe the bug
When attempting to import the models module from sagemaker.jumpstart, the following error occurs:

ImportError: cannot import name 'models' from 'sagemaker.jumpstart' (/opt/conda/lib/python3.11/site-packages/sagemaker/jumpstart/__init__.py)

To reproduce

from sagemaker.jumpstart import models

def confirm_model_availability(model_id):
    available_models = models.list_models()

Expected behavior
The models module should be successfully imported from sagemaker.jumpstart.

Actual Behavior
The import fails with an ImportError.

Screenshots or logs

System information
A description of your system. Please provide:

• SageMaker Python SDK version: 2.237.1
• Framework name (eg. PyTorch) or algorithm (eg. KMeans): I was using Jupyter Lab in Sagemaker studio
• Python version: 3.11.10

Additional context
It was working a week ago.