Can't install sagemaker library due to incompatibilities

[pcolazurdo]

Trying to run pip install sagemaker as of vv2.198.1 produces the following error:

ERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts.
sagemaker-sklearn-container 2.0 requires boto3==1.28.57, but you have boto3 1.34.32 which is incompatible.
sagemaker-sklearn-container 2.0 requires botocore<1.32.0,>=1.31.57, but you have botocore 1.34.32 which is incompatible.

As a workaround, going back to version sagemaker==2.197.0 seems to work, but this disable some new features.