marintosti12 commited on
Commit
09ccbd8
·
1 Parent(s): 84394db

fix(loader) : fix env loader

Browse files
Files changed (2) hide show
  1. .github/workflows/ci.yaml +2 -0
  2. src/model_loader.py +1 -1
.github/workflows/ci.yaml CHANGED
@@ -18,6 +18,8 @@ jobs:
18
  POSTGRES_USER: ci
19
  POSTGRES_PASSWORD: ci
20
  POSTGRES_DB: futurisys_ci
 
 
21
  ports:
22
  - 5432:5432
23
  options: >-
 
18
  POSTGRES_USER: ci
19
  POSTGRES_PASSWORD: ci
20
  POSTGRES_DB: futurisys_ci
21
+ APP_ENV: test
22
+
23
  ports:
24
  - 5432:5432
25
  options: >-
src/model_loader.py CHANGED
@@ -22,7 +22,7 @@ def _load_local(name: str) -> Any:
22
  @lru_cache(maxsize=1)
23
  def load_model(name) -> Any:
24
 
25
- if ENV in ("dev", "test"):
26
  return _load_local(name)
27
 
28
  hf_path = hf_hub_download(
 
22
  @lru_cache(maxsize=1)
23
  def load_model(name) -> Any:
24
 
25
+ if ENV in ("dev"):
26
  return _load_local(name)
27
 
28
  hf_path = hf_hub_download(