From ded6e833239ea1cfb27f9df6d0fd390ca0adecbd Mon Sep 17 00:00:00 2001 From: Chris McKenzie Date: Wed, 22 Jan 2025 11:29:57 -0800 Subject: [PATCH] vast: ssh port code fix --- sky/provision/vast/utils.py | 2 +- tests/smoke_tests/test_basic.py | 3 +-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/sky/provision/vast/utils.py b/sky/provision/vast/utils.py index d483950a929..250bf61df55 100644 --- a/sky/provision/vast/utils.py +++ b/sky/provision/vast/utils.py @@ -136,7 +136,7 @@ def get_ssh_ports(cluster_name: str) -> List[int]: for instance in instances.values(): if instance['name'] in possible_names: - ssh_ports.append((instance['name'], instance['ssh_port'])) + ssh_ports.append(instance['ssh_port']) assert ssh_ports, ( f'Could not find any instances for cluster {cluster_name}.') diff --git a/tests/smoke_tests/test_basic.py b/tests/smoke_tests/test_basic.py index e837e0cf145..e5b2ed14da9 100644 --- a/tests/smoke_tests/test_basic.py +++ b/tests/smoke_tests/test_basic.py @@ -148,8 +148,7 @@ def test_launch_fast_with_autostop(generic_cloud: str): f'sky status -r {name} | grep UP', ], f'sky down -y {name}', - timeout=smoke_tests_utils.get_timeout(generic_cloud) + - autostop_timeout, + timeout=smoke_tests_utils.get_timeout(generic_cloud) + autostop_timeout, ) smoke_tests_utils.run_one_test(test)