Fix fxn.predictions.stream

fxnai · Apr 9, 2024 · c70a8c9 · c70a8c9
1 parent 17b6463
commit c70a8c9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git 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
@@ -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