fix type check

Signed-off-by: Alex Chi Z <[email protected]>
neondatabase · Nov 19, 2024 · cdde254 · cdde254
1 parent a9db766
commit cdde254
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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}",
         )