junte/Model
1
0
Fork 0
mirror of https://github.com/Hestia-Homes/Model.git synced 2026-06-08 11:17:27 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

lambda can only write to /tmp folder to using /tmp for model download

Browse source
This commit is contained in:
Khalim Conn-Kowlessar 2023-09-05 12:43:51 +01:00
parent 244665c27f
commit 66b20c2223
2 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
model_data/simulation_system/MLModel/Models.py
View file

@ -66,7 +66,7 @@ class AutogluonModel:
logger.info("In local development mode - no need for s3 client")
self.model = TabularPredictor.load(path=filepath)
else:
logger.info(f"Loading model from s3")
logger.info(f"Loading model from s3 with filepath: %s and model_folder: %s" % (filepath, model_folder))
client.download_model(filepath=filepath, model_folder=model_folder)
self.model = TabularPredictor.load(path=str(Path(model_folder) / filepath))

5
model_data/simulation_system/predictions.py
View file

@ -135,7 +135,10 @@ def prediction(
else:
raise ValueError("No other model currently")
model.load_model(filepath=model_location, client=CLIENT, model_folder="local_model")
# In lambda, only the /tmp folder is writable
model_folder = "/tmp" if RUNTIME_ENVIRONMENT in ["dev", "prod"] else "local_model"
model.load_model(filepath=model_location, client=CLIENT, model_folder=model_folder)
logger.info("--- Generating Predictions ---")
prediction = model.generate_predictions(data=data)