diff --git a/tests/integ/test_sklearn.py b/tests/integ/test_sklearn.py index 839e601d34..ff5b466b3f 100644 --- a/tests/integ/test_sklearn.py +++ b/tests/integ/test_sklearn.py @@ -159,8 +159,6 @@ def test_deploy_model( def test_deploy_model_with_serverless_inference_config( sklearn_training_job, sagemaker_session, - sklearn_latest_version, - sklearn_latest_py_version, ): endpoint_name = unique_name_from_base("test-sklearn-deploy-model-serverless") with timeout_and_delete_endpoint_by_name(endpoint_name, sagemaker_session): @@ -173,7 +171,7 @@ def test_deploy_model_with_serverless_inference_config( model_data, ROLE, entry_point=script_path, - framework_version=sklearn_latest_version, + framework_version="1.0-1", sagemaker_session=sagemaker_session, ) predictor = model.deploy( diff --git a/tests/integ/test_xgboost.py b/tests/integ/test_xgboost.py index 7b4db837fd..1c06c6b5c6 100644 --- a/tests/integ/test_xgboost.py +++ b/tests/integ/test_xgboost.py @@ -121,11 +121,9 @@ def test_training_with_network_isolation( ] -@pytest.mark.skip(reason="re:Invent keynote3 blocker. Revisit after release") def test_xgboost_serverless_inference( xgboost_training_job, sagemaker_session, - xgboost_latest_version, ): endpoint_name = unique_name_from_base("test-xgboost-deploy-model-serverless") with timeout_and_delete_endpoint_by_name(endpoint_name, sagemaker_session): @@ -139,7 +137,7 @@ def test_xgboost_serverless_inference( model_data=model_data, role=ROLE, entry_point=os.path.join(DATA_DIR, "xgboost_abalone", "abalone.py"), - framework_version=xgboost_latest_version, + framework_version="1.5-1", ) xgboost.deploy(