From c70a8c9021eb7808d6b84e752359f2ffb3119367 Mon Sep 17 00:00:00 2001 From: Yusuf Olokoba 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