fix type check

Signed-off-by: Alex Chi Z <chi@neon.tech>
This commit is contained in:
Alex Chi Z
2024-11-19 11:58:06 -05:00
parent a9db766c20
commit cdde254c84

View File

@@ -346,7 +346,7 @@ class PageserverHttpClient(requests.Session, MetricsGetter):
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}",
)