diff --git a/test_runner/fixtures/pageserver/http.py b/test_runner/fixtures/pageserver/http.py index d1a9b5921a2b..080a80c0e4cf 100644 --- a/test_runner/fixtures/pageserver/http.py +++ b/test_runner/fixtures/pageserver/http.py @@ -346,7 +346,7 @@ def tenant_list_locations(self): assert isinstance(res_json["tenant_shards"], list) return res_json - def tenant_get_location(self, tenant_id: TenantShardId): + def tenant_get_location(self, tenant_id: Union[TenantId, TenantShardId]): res = self.get( f"http://localhost:{self.port}/v1/location_config/{tenant_id}", )