From c70a8c9021eb7808d6b84e752359f2ffb3119367 Mon Sep 17 00:00:00 2001
From: Yusuf Olokoba <olokobayusuf@gmail.com>
Date: Tue, 9 Apr 2024 14:46:56 -0400
Subject: [PATCH] Fix `fxn.predictions.stream`

---
 Changelog.md                       | 2 +-
 fxn/services/prediction/service.py | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/Changelog.md b/Changelog.md
index 755b6db..94758b2 100644
--- a/Changelog.md
+++ b/Changelog.md
@@ -1,5 +1,5 @@
 ## 0.0.34
-*INCOMPLETE*
++ Fixed `fxn` import error caused by `fxn.predictions.stream` function.
 
 ## 0.0.33
 + Fixed error in `fxn.predictors.retrieve` function.
diff --git a/fxn/services/prediction/service.py b/fxn/services/prediction/service.py
index 24b096c..2dc02f7 100644
--- a/fxn/services/prediction/service.py
+++ b/fxn/services/prediction/service.py
@@ -155,7 +155,8 @@ async def stream (
                     prediction = self.__parse_prediction(prediction, raw_outputs=raw_outputs, return_binary_path=return_binary_path)
                     # Check edge prediction
                     if prediction.type != PredictorType.Edge or raw_outputs:
-                        return prediction
+                        yield prediction
+                        continue
                     # Load edge predictor
                     predictor = self.__load(prediction)
                     self.__cache[tag] = predictor