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"""