dvarfe commited on
Commit
5f75018
·
1 Parent(s): 2603501

fix(model catalog): add feather fallback

Browse files
Files changed (2) hide show
  1. Dockerfile +2 -2
  2. dashboard/model_catalog.py +4 -1
Dockerfile CHANGED
@@ -13,7 +13,7 @@ RUN apt-get update && apt-get install -y \
13
  bash \
14
  && rm -rf /var/lib/apt/lists/*
15
 
16
- RUN useradd -m -u 1000 dvarfe
17
 
18
  WORKDIR /app
19
 
@@ -23,7 +23,7 @@ RUN pip install --no-cache-dir --upgrade -r requirements.txt
23
 
24
  COPY --chown=1000:1000 --link . /app
25
 
26
- ENV PATH="/home/dvarfe/.local/bin:$PATH"
27
  ENV DASH_PORT=7860
28
  ENV MODEL_STORAGE='/data/logs'
29
 
 
13
  bash \
14
  && rm -rf /var/lib/apt/lists/*
15
 
16
+ RUN useradd -m -u 1000 user
17
 
18
  WORKDIR /app
19
 
 
23
 
24
  COPY --chown=1000:1000 --link . /app
25
 
26
+ ENV PATH="/home/user/.local/bin:$PATH"
27
  ENV DASH_PORT=7860
28
  ENV MODEL_STORAGE='/data/logs'
29
 
dashboard/model_catalog.py CHANGED
@@ -440,7 +440,10 @@ def load_selector_cache_feature_table(cache_path_str: str) -> pd.DataFrame:
440
  if not path.exists():
441
  return pd.DataFrame(columns=columns)
442
 
443
- table = pd.read_parquet(path)
 
 
 
444
  numeric = table.select_dtypes(include=["number"])
445
  if numeric.empty:
446
  return pd.DataFrame(columns=columns)
 
440
  if not path.exists():
441
  return pd.DataFrame(columns=columns)
442
 
443
+ try:
444
+ table = pd.read_parquet(path)
445
+ except Exception as e:
446
+ table = pd.read_feather(path.with_suffix('.feather')) # Hugging Face Parquet patch
447
  numeric = table.select_dtypes(include=["number"])
448
  if numeric.empty:
449
  return pd.DataFrame(columns=columns)