[torchx/dev-requirements] Fix broken unittest in CI caused by new releases of sagemaker (2.235.2) and mlflow-skinny (2.18)

Author: Kiuk Chung

dev-requirements.txt

@@ -14,17 +14,19 @@ google-cloud-runtimeconfig==0.34.0
 hydra-core
 ipython
 kfp==1.8.22
-mlflow-skinny
+mlflow-skinny<=2.17.2
 moto~=5.0.8
 pyre-extensions
 pyre-check
 pytest
 pytest-cov
 pytorch-lightning==2.3.1
 tensorboard==2.14.0
-sagemaker>=2.149.0
+# sagemaker>=2.235.2 seems to be a bad release. APIs fail with
+# NameError: Field name "json" shadows a BaseModel attribute; use a different field name with "alias='json'"
+sagemaker>=2.149.0,<=2.235.1
 torch-model-archiver>=0.4.2
-torch==2.2.1
+torch>=2.2.1
 torchmetrics==0.10.3
 torchserve>=0.10.0
 torchtext==0.17.1
@@ -42,5 +44,4 @@ lintrunner-adapters
 grpcio==1.62.1
 grpcio-status==1.48.1
 googleapis-common-protos==1.63.0
-google-api-core==2.18.0
 protobuf==3.20.3 # kfp==1.8.22 needs protobuf < 4