Import cohere_ws giving an issue No module named 'sagemaker.s3'

[mnprabhadevi-create]

@alex-matton @jpekmez

import cohere_aws
Traceback (most recent call last):
File "", line 1, in
File "/home/ubuntu/AWS_learnings/cohere-aws/cohere_aws/init.py", line 1, in
from .client import Client
File "/home/ubuntu/AWS_learnings/cohere-aws/cohere_aws/client.py", line 12, in
from sagemaker.s3 import S3Downloader, S3Uploader, parse_s3_url
ModuleNotFoundError: No module named 'sagemaker.s3'

[robertkerr-cohere]

This is caused by using sagemaker version 3, which removed the sagemaker.s3 path. This package should pin sagemaker<3 or upgrade code for v3.