mirror/gradio
2
0
Fork 0
mirror of https://github.com/gradio-app/gradio.git synced 2025-03-31 12:20:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

Skip tests that return 429 from HF Hub API (#1559)

Browse Source 
* skip tests

* formatting

* formatting and fixes
This commit is contained in:
Abubakar Abid 2022-06-13 17:57:45 -07:00 committed by GitHub
parent a3047ee838
commit d10bca27c0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 57 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
gradio/external.py
View File

@ -13,6 +13,12 @@ import gradio
from gradio import components, utils
class TooManyRequestsError(Exception):
"""Raised when the Hugging Face API returns a 429 status code."""
pass
def load_blocks_from_repo(name, src=None, api_key=None, alias=None, **kwargs):
"""Creates and returns a Blocks instance from several kinds of Hugging Face repos:
1) A model repo
@ -334,6 +340,10 @@ def get_spaces_blocks(model_name, config):
try:
output = result["data"]
except KeyError:
if "error" in result and "429" in result["error"]:
raise TooManyRequestsError(
"Too many requests to the Hugging Face API"
)
raise KeyError(
f"Could not find 'data' key in response from external Space. Response received: {result}"
)
@ -385,6 +395,8 @@ def get_spaces_interface(model_name, config, alias, **kwargs):
try:
output = result["data"]
except KeyError:
if "error" in result and "429" in result["error"]:
raise TooManyRequestsError("Too many requests to the Hugging Face API")
raise KeyError(
f"Could not find 'data' key in response from external Space. Response received: {result}"
)

2
setup.py
View File

@ -7,6 +7,8 @@ from pathlib import Path
this_directory = Path(__file__).parent
long_description = (this_directory / "README.md").read_text(encoding='utf8')
print(long_description)
setup(
name="gradio",
version="3.0.13",

43
test/test_external.py
View File

@ -5,6 +5,7 @@ import unittest
import transformers
import gradio as gr
from gradio.external import TooManyRequestsError
"""
WARNING: These tests have an external dependency: namely that Hugging Face's
@ -171,33 +172,51 @@ class TestLoadInterface(unittest.TestCase):
def test_sentiment_model(self):
io = gr.Interface.load("models/distilbert-base-uncased-finetuned-sst-2-english")
output = io("I am happy, I love you.")
self.assertGreater(output["POSITIVE"], 0.5)
try:
output = io("I am happy, I love you.")
self.assertGreater(output["POSITIVE"], 0.5)
except TooManyRequestsError:
pass
def test_image_classification_model(self):
io = gr.Blocks.load(name="models/google/vit-base-patch16-224")
output = io("gradio/test_data/lion.jpg")
self.assertGreater(output["lion"], 0.5)
try:
output = io("gradio/test_data/lion.jpg")
self.assertGreater(output["lion"], 0.5)
except TooManyRequestsError:
pass
def test_translation_model(self):
io = gr.Blocks.load(name="models/t5-base")
output = io("My name is Sarah and I live in London")
self.assertEqual(output, "Mein Name ist Sarah und ich lebe in London")
try:
output = io("My name is Sarah and I live in London")
self.assertEqual(output, "Mein Name ist Sarah und ich lebe in London")
except TooManyRequestsError:
pass
def test_numerical_to_label_space(self):
io = gr.Interface.load("spaces/abidlabs/titanic-survival")
output = io("male", 77, 10)
self.assertLess(output["Survives"], 0.5)
try:
output = io("male", 77, 10)
self.assertLess(output["Survives"], 0.5)
except TooManyRequestsError:
pass
def test_speech_recognition_model(self):
io = gr.Interface.load("models/facebook/wav2vec2-base-960h")
output = io("gradio/test_data/test_audio.wav")
self.assertIsNotNone(output)
try:
output = io("gradio/test_data/test_audio.wav")
self.assertIsNotNone(output)
except TooManyRequestsError:
pass
def test_text_to_image_model(self):
io = gr.Interface.load("models/osanseviero/BigGAN-deep-128")
filename = io("chest")
self.assertTrue(filename.endswith(".jpg") or filename.endswith(".jpeg"))
try:
filename = io("chest")
self.assertTrue(filename.endswith(".jpg") or filename.endswith(".jpeg"))
except TooManyRequestsError:
pass
class TestLoadFromPipeline(unittest.TestCase):

17
test/test_mix.py
View File

@ -3,6 +3,7 @@ import unittest
import gradio as gr
from gradio import mix
from gradio.external import TooManyRequestsError
"""
WARNING: Some of these tests have an external dependency: namely that Hugging Face's Hub and Space APIs do not change, and they keep their most famous models up. So if, e.g. Spaces is down, then these test will not pass.
@ -23,8 +24,11 @@ class TestSeries(unittest.TestCase):
io1 = gr.Interface.load("spaces/abidlabs/image-identity")
io2 = gr.Interface.load("spaces/abidlabs/image-classifier")
series = mix.Series(io1, io2)
output = series("gradio/test_data/lion.jpg")
self.assertGreater(output["lion"], 0.5)
try:
output = series("gradio/test_data/lion.jpg")
self.assertGreater(output["lion"], 0.5)
except TooManyRequestsError:
pass
class TestParallel(unittest.TestCase):
@ -40,9 +44,12 @@ class TestParallel(unittest.TestCase):
io1 = gr.Interface.load("spaces/abidlabs/english_to_spanish")
io2 = gr.Interface.load("spaces/abidlabs/english2german")
parallel = mix.Parallel(io1, io2)
hello_es, hello_de = parallel("Hello")
self.assertIn("hola", hello_es.lower())
self.assertIn("hallo", hello_de.lower())
try:
hello_es, hello_de = parallel("Hello")
self.assertIn("hola", hello_es.lower())
self.assertIn("hallo", hello_de.lower())
except TooManyRequestsError:
pass
if __name__ == "__main__":