From 80248a117830d40e2c2b1f2a8c03819cd8be4395 Mon Sep 17 00:00:00 2001 From: ewezy Date: Mon, 20 May 2024 17:49:34 +0800 Subject: [PATCH] Fix integration tests --- python/sdk/test/integration_test.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/python/sdk/test/integration_test.py b/python/sdk/test/integration_test.py index 3f4fc0dfe..c6d4791ea 100644 --- a/python/sdk/test/integration_test.py +++ b/python/sdk/test/integration_test.py @@ -504,7 +504,7 @@ def test_resource_request( # Upload the serialized model to MLP merlin.log_model(model_dir=model_dir) - resource_request = ResourceRequest(1, 1, "100m", "256Mi") + resource_request = ResourceRequest(1, 1, "100m", "2", "256Mi") endpoint = merlin.deploy( v, environment_name=default_env.name, @@ -560,7 +560,7 @@ def test_resource_request_with_gpu( # Upload the serialized model to MLP merlin.log_model(model_dir=model_dir) - resource_request = ResourceRequest(1, 1, "100m", "256Mi", **gpu_config) + resource_request = ResourceRequest(1, 1, "100m", "2", "256Mi", **gpu_config) endpoint = merlin.deploy( v, environment_name=default_env.name, @@ -1271,7 +1271,7 @@ def test_redeploy_model(integration_test_url, project_name, use_google_oauth, re ), deployment_mode=DeploymentMode.RAW_DEPLOYMENT, transformer=Transformer(image="gcr.io/kubeflow-ci/kfserving/image-transformer:latest", - resource_request=merlin.ResourceRequest(0, 1, "100m", "3", "250Mi")), + resource_request=merlin.ResourceRequest(0, 1, "100m", "2", "250Mi")), ) resp = requests.post(f"{new_endpoint.url}", json=req)