diff --git a/backend/ml_models/api.py b/backend/ml_models/api.py index 6c92df2a..02d6ecac 100644 --- a/backend/ml_models/api.py +++ b/backend/ml_models/api.py @@ -56,10 +56,7 @@ class ModelApi: raise ValueError(f"Model prefix specified is not in {self.MODEL_PREFIXES}") # Store parquet file in s3 for scoring - file_location = "sap_change_predictions/{portfolio_id}/{timestamp}.parquet".format( - portfolio_id=self.portfolio_id, - timestamp=self.timestamp - ) + file_location = f"{model_prefix}/{self.portfolio_id}/{self.timestamp}.parquet" logger.info("Storing scoring data to s3") save_dataframe_to_s3_parquet(