updated save filetype to parquet

This commit is contained in:
Khalim Conn-Kowlessar 2023-10-03 18:23:51 +01:00
parent e4352bda1e
commit 5e62b2d43e

View file

@ -67,9 +67,7 @@ def handler(event, context):
# TODO: Implement the loading of the model and prediction
storage_filepath = (
f"s3://{PREDICTIONS_BUCKET}/{portfolio_id}/{property_id}/{created_at}.csv"
)
storage_filepath = f"s3://{PREDICTIONS_BUCKET}/{portfolio_id}/{property_id}/{created_at}.parquet"
logger.info("-------------------------")
logger.info(f"--- Initiate MLModel ---")