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