alessandro trinca tornidor commited on
Commit
c1ebf60
1 Parent(s): 5d4e838

[feat] get machine_learning_models folder from MODEL_FOLDER env var

Browse files
Files changed (1) hide show
  1. samgis_lisa_on_cuda/__init__.py +2 -1
samgis_lisa_on_cuda/__init__.py CHANGED
@@ -9,7 +9,8 @@ from samgis_lisa_on_cuda.utilities.constants import SERVICE_NAME
9
  ROOT = Path(globals().get("__file__", "./_")).absolute().parent.parent
10
  PROJECT_ROOT_FOLDER = Path(os.getenv("PROJECT_ROOT_FOLDER", ROOT))
11
  WORKDIR = Path(os.getenv("WORKDIR", ROOT))
12
- MODEL_FOLDER = Path(PROJECT_ROOT_FOLDER / "machine_learning_models")
 
13
 
14
  IS_AWS_LAMBDA = bool(os.getenv("IS_AWS_LAMBDA", ""))
15
 
 
9
  ROOT = Path(globals().get("__file__", "./_")).absolute().parent.parent
10
  PROJECT_ROOT_FOLDER = Path(os.getenv("PROJECT_ROOT_FOLDER", ROOT))
11
  WORKDIR = Path(os.getenv("WORKDIR", ROOT))
12
+ MODEL_FOLDER_PROJECT_ROOT_FOLDER = Path(PROJECT_ROOT_FOLDER / "machine_learning_models")
13
+ MODEL_FOLDER = Path(os.getenv("MODEL_FOLDER", MODEL_FOLDER_PROJECT_ROOT_FOLDER))
14
 
15
  IS_AWS_LAMBDA = bool(os.getenv("IS_AWS_LAMBDA", ""))
16