From 92da069f57ee0c87a937d2c3b1ecbfbb240fc6e8 Mon Sep 17 00:00:00 2001 From: Zhanghao Wu Date: Tue, 6 Aug 2024 09:33:05 -0700 Subject: [PATCH] [Tests] Fix newly added CPU instance for lambda (#3811) newly added CPU instance --- tests/test_optimizer_dryruns.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_optimizer_dryruns.py b/tests/test_optimizer_dryruns.py index 3d557ea475f..becf3ba461a 100644 --- a/tests/test_optimizer_dryruns.py +++ b/tests/test_optimizer_dryruns.py @@ -274,7 +274,7 @@ def test_instance_type_from_cpu_memory(monkeypatch, capfd): assert 'n2-highcpu-4' in stdout # GCP, 4 vCPUs, 4 GB memory assert 'c6i.xlarge' in stdout # AWS, 4 vCPUs, 8 GB memory assert 'Standard_F4s_v2' in stdout # Azure, 4 vCPUs, 8 GB memory - assert 'gpu_1x_rtx6000' in stdout # Lambda, 14 vCPUs, 46 GB memory + assert 'cpu_4x_general' in stdout # Lambda, 4 vCPUs, 16 GB memory _test_resources_launch(monkeypatch, accelerators='T4') stdout, _ = capfd.readouterr()