diff --git a/tests/common.py b/tests/common.py index 835f53ef251..4844def6bb9 100644 --- a/tests/common.py +++ b/tests/common.py @@ -55,7 +55,7 @@ def _get_az_mappings(_): # the cluster to detect available cluster resources. monkeypatch.setattr( 'sky.utils.kubernetes_utils.detect_gpu_label_formatter', - lambda *_args, **_kwargs: [kubernetes_utils.SkyPilotLabelFormatter, []]) + lambda *_args, **_kwargs: [kubernetes_utils.SkyPilotLabelFormatter, {}]) monkeypatch.setattr('sky.utils.kubernetes_utils.detect_gpu_resource', lambda *_args, **_kwargs: [True, []]) monkeypatch.setattr('sky.utils.kubernetes_utils.check_instance_fits',