fixed reference to predictions

This commit is contained in:
Khalim Conn-Kowlessar 2023-09-01 10:46:17 +01:00
parent f8209c42ab
commit fe6a08f031
2 changed files with 6 additions and 2 deletions

View file

@ -3,6 +3,9 @@ FROM public.ecr.aws/lambda/python:3.10
# Set the working directory
WORKDIR ${LAMBDA_TASK_ROOT}
# Install necessary build tools - required to test locally
RUN yum install -y gcc python3-devel
# Install python packages
COPY requirements/predictions/predictions.txt ./simulation_system/requirements.txt
RUN pip install --no-cache-dir -r ./simulation_system/requirements.txt

View file

@ -1,6 +1,6 @@
import os
import urllib.parse
from predictions import prediction
from ..predictions import prediction
RUNTIME_ENVIRONMENT = os.environ.get("RUNTIME_ENVIRONMENT", "dev")
@ -30,7 +30,8 @@ def handler(event, context):
# model_path = os.environ.get("MODEL_PATH", "http://minio:9000/data/model_directory/")
model_path = os.environ.get(
"MODEL_PATH",
"s3://retrofit-model-directory-{RUNTIME_ENVIRONMENT}/RDSAP_CHANGE/autogluon/rdsap_change-medium_quality-30-2023-08-30_11-43-41/deployment/",
"s3://retrofit-model-directory-{RUNTIME_ENVIRONMENT}/RDSAP_CHANGE/autogluon/rdsap_change-medium_quality-30"
"-2023-08-30_11-43-41/deployment/",
)
try: