Remove external data download — curated tasks embedded in code

Dockerfile

@@ -49,15 +49,10 @@ RUN apt-get update && \
 COPY --from=builder /app/env/.venv /app/.venv
 COPY --from=builder /app/env /app/env
 
-# Extract shaders21k corpus
-# Priority: bundled zip > already extracted > download from HF Space
+# Extract shaders21k corpus if bundled; otherwise curated tasks suffice
 RUN if [ -f /app/env/shader_data.zip ]; then \
         python3 -c "import zipfile; zipfile.ZipFile('/app/env/shader_data.zip').extractall('/app/env')" && \
         rm /app/env/shader_data.zip; \
-    elif [ ! -d /app/env/data/shadertoy ]; then \
-        curl -fSL "https://huggingface.co/spaces/tejadhith/shader/resolve/main/shader_data.zip" -o /tmp/shader_data.zip && \
-        python3 -c "import zipfile; zipfile.ZipFile('/tmp/shader_data.zip').extractall('/app/env')" && \
-        rm /tmp/shader_data.zip; \
     fi
 
 ENV PATH="/app/.venv/bin:$PATH"

server/Dockerfile

@@ -0,0 +1,64 @@
+# Multi-stage build for the shader environment.
+# Requires EGL for headless OpenGL rendering.
+
+ARG BASE_IMAGE=ghcr.io/meta-pytorch/openenv-base:latest
+FROM ${BASE_IMAGE} AS builder
+
+WORKDIR /app
+
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends git && \
+    rm -rf /var/lib/apt/lists/*
+
+ARG BUILD_MODE=in-repo
+ARG ENV_NAME=shader
+
+COPY . /app/env
+
+WORKDIR /app/env
+
+RUN if ! command -v uv >/dev/null 2>&1; then \
+        curl -LsSf https://astral.sh/uv/install.sh | sh && \
+        mv /root/.local/bin/uv /usr/local/bin/uv && \
+        mv /root/.local/bin/uvx /usr/local/bin/uvx; \
+    fi
+
+RUN if [ -f uv.lock ]; then \
+        uv sync --frozen --no-install-project --no-editable; \
+    else \
+        uv sync --no-install-project --no-editable; \
+    fi
+
+RUN if [ -f uv.lock ]; then \
+        uv sync --frozen --no-editable; \
+    else \
+        uv sync --no-editable; \
+    fi
+
+# Final runtime stage
+FROM ${BASE_IMAGE}
+
+WORKDIR /app
+
+# EGL support for headless OpenGL rendering
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends \
+        libegl1 libgl1 libgles2 libgl1-mesa-dri && \
+    rm -rf /var/lib/apt/lists/*
+
+COPY --from=builder /app/env/.venv /app/.venv
+COPY --from=builder /app/env /app/env
+
+# Extract shaders21k corpus if bundled; otherwise curated tasks suffice
+RUN if [ -f /app/env/shader_data.zip ]; then \
+        python3 -c "import zipfile; zipfile.ZipFile('/app/env/shader_data.zip').extractall('/app/env')" && \
+        rm /app/env/shader_data.zip; \
+    fi
+
+ENV PATH="/app/.venv/bin:$PATH"
+ENV PYTHONPATH="/app/env:$PYTHONPATH"
+
+HEALTHCHECK --interval=30s --timeout=3s --start-period=5s --retries=3 \
+    CMD curl -f http://localhost:8000/health || exit 1
+
+CMD ["sh", "-c", "cd /app/env && uvicorn server.app:app --host 0.0.0.0 --port 8000"]

server/environment.py

@@ -40,7 +40,11 @@ class ShaderEnvironment(Environment):
         self._budget = budget
         self._rng = random.Random(seed)
         if ShaderEnvironment._tasks is None:
-            ShaderEnvironment._tasks = load_tasks()
+            try:
+                ShaderEnvironment._tasks = load_tasks()
+            except FileNotFoundError:
+                # Corpus not available — use curated tasks only
+                ShaderEnvironment._tasks = list(CURATED)
         self._state = State(episode_id=None, step_count=0)
 
         # Episode state