From 746ba8bb604e21ac4f600ad8840effecc6cbde9a Mon Sep 17 00:00:00 2001 From: kunato Date: Sun, 21 May 2023 22:51:14 +0700 Subject: [PATCH] fix iterator has override output --- replicate/prediction.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/replicate/prediction.py b/replicate/prediction.py index a69217d8..ee164fb8 100644 --- a/replicate/prediction.py +++ b/replicate/prediction.py @@ -33,8 +33,8 @@ def output_iterator(self) -> Iterator[Any]: while self.status not in ["succeeded", "failed", "canceled"]: output = self.output or [] new_output = output[len(previous_output) :] - for output in new_output: - yield output + for o in new_output: + yield o previous_output = output time.sleep(self._client.poll_interval) self.reload() @@ -44,8 +44,8 @@ def output_iterator(self) -> Iterator[Any]: output = self.output or [] new_output = output[len(previous_output) :] - for output in new_output: - yield output + for o in new_output: + yield o def cancel(self) -> None: """Cancel a currently running prediction"""