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

Fixes #7136 -- gradio client iterator can hit finish event before yield all outputs (#7151)

Browse Source
This commit is contained in:
PSEUDOTENSOR / Jonathan McKinney 2024-01-25 16:57:30 -08:00 committed by GitHub
parent 138761226a
commit 40e6ddf911
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
client/python/gradio_client/client.py
View File

@ -1557,7 +1557,10 @@ class Job(Future):
o = self.communicator.job.outputs[self._counter]
self._counter += 1
return o
if self.communicator.job.latest_status.code == Status.FINISHED:
if (
self.communicator.job.latest_status.code == Status.FINISHED
and self._counter >= len(self.communicator.job.outputs)
):
raise StopIteration()
time.sleep(0.001)