fix flaky tests

gradio-app · Oct 5, 2024 · 4388ee3 · 4388ee3
1 parent b82aa6f
commit 4388ee3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/client/python/gradio_client/client.py b/client/python/gradio_client/client.py
@@ -1080,7 +1080,7 @@ def __init__(
         ]
         self.parameters_info = self._get_parameters_info()
 
-        self.root_url = client.src.rstrip("/") + "/" + client.api_prefix
+        self.root_url = self.client.src_prefixed
 
         # Disallow hitting endpoints that the Gradio app has disabled
         self.is_valid = self.api_name is not False

diff --git a/client/python/test/test_client.py b/client/python/test/test_client.py
@@ -130,7 +130,7 @@ def test_private_space(self):
         space_id = "gradio-tests/not-actually-private-space"
         api = huggingface_hub.HfApi()
         assert api.space_info(space_id).private
-        client = Client(space_id)
+        client = Client(space_id, hf_token=None)
         output = client.predict("abc", api_name="/predict")
         assert output == "abc"
 
@@ -141,6 +141,7 @@ def test_private_space_v4(self):
         assert api.space_info(space_id).private
         client = Client(
             space_id,
+            hf_token=None,
         )
         output = client.predict("abc", api_name="/predict")
         assert output == "abc"
@@ -152,6 +153,7 @@ def test_private_space_v4_sse_v1(self):
         assert api.space_info(space_id).private
         client = Client(
             space_id,
+            hf_token=None,
         )
         output = client.predict("abc", api_name="/predict")
         assert output == "abc"