Change model_base_dir to point to existing folder in repo

beeldengeluid · Oct 8, 2024 · 87d569e · 87d569e
1 parent b0c753a
commit 87d569e
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 21 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -15,7 +15,6 @@ boto3 = "^1.35.10"
 fastapi = "^0.115.0"
 uvicorn = "^0.30.6"
 tomli = "^2.0.1"
-python-dotenv = "^1.0.1"
 
 [tool.poetry.group.dev.dependencies]
 moto = "^5.0.13"

diff --git a/tests/.env b/tests/.env
diff --git a/tests/test_model_download.py b/tests/test_model_download.py
@@ -1,10 +1,11 @@
 import pytest
 import shutil
 import os
-from dotenv import load_dotenv
 
 # Mocking environment used in model_download (by loading tests/.env)
-load_dotenv()
+os.environ["DATA_BASE_DIR"] = "data"
+os.environ["MODEL_BASE_DIR"] = "tests/input/extract_model_test"
+os.environ["S3_ENDPOINT_URL"] = "http://url.com"
 
 from model_download import extract_model, get_model_location  # noqa