Use AsyncStopIteration

gradio-app · Oct 7, 2022 · 41d39f9 · 41d39f9
1 parent f812365
commit 41d39f9
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 6 deletions.
diff --git a/gradio/blocks.py b/gradio/blocks.py
@@ -707,7 +707,7 @@ async def call_function(self, fn_index, processed_input, iterator=None):
                     utils.async_iteration, iterator, limiter=self.limiter
                 )
                 is_generating = True
-            except GradioStopIteration:
+            except StopAsyncIteration:
                 n_outputs = len(self.dependencies[fn_index].get("outputs"))
                 prediction = (
                     components._Keywords.FINISHED_ITERATING

diff --git a/gradio/exceptions.py b/gradio/exceptions.py
@@ -9,7 +9,3 @@ def __init__(self, message: str):
 
     def __str__(self):
         return repr(self.message)
-
-
-class GradioStopIteration(ValueError):
-    pass
diff --git a/gradio/utils.py b/gradio/utils.py
@@ -389,7 +389,7 @@ def async_iteration(iterator):
         return next(iterator)
     except StopIteration:
         # raise a ValueError here because co-routines can't raise StopIteration themselves
-        raise gradio.exceptions.GradioStopIteration()
+        raise StopAsyncIteration()
 
 
 class Request: