Spaces:
Running
Running
fix(model catalog): add feather fallback
Browse files- Dockerfile +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
|
| 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/
|
| 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 |
-
|
|
|
|
|
|
|
|
|
|
| 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)
|