Skip to content

Commit 335bc06

Browse files
authored
Merge pull request #168 from datakind/AdjustedModelCardsEnv
added case block for env definition
2 parents 25b447c + f711a17 commit 335bc06

1 file changed

Lines changed: 2 additions & 3 deletions

File tree

src/webapp/routers/data.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
import uuid
44
from datetime import datetime, date
55
from databricks.sdk import WorkspaceClient
6-
from typing import Annotated, Any, Dict, List, cast, IO, Optional, Tuple, Sequence
6+
from typing import Annotated, Any, Dict, List, cast, IO, Optional, Tuple
77
from pydantic import BaseModel, Field
88
from fastapi import APIRouter, Depends, HTTPException, status, Response, Query
99
from fastapi.responses import FileResponse
@@ -1771,8 +1771,7 @@ def get_model_cards(
17711771
)
17721772

17731773
try:
1774-
env_vals: Sequence[str] = env_vars.get("ENV", [])
1775-
env = (env_vals[0] if env_vals else "").strip().upper()
1774+
env = str(env_vars["ENV"]).strip().upper()
17761775
SCHEMAS = {"DEV": "dev_sst_02", "STAGING": "staging_sst_01"}
17771776
if env not in SCHEMAS:
17781777
raise ValueError(

0 commit comments

Comments
 (0)