diff --git a/Dockerfile b/Dockerfile index 05ce8e3..a5823aa 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,12 +1,5 @@ FROM python:3.11-slim -ARG TASK_TYPE -ARG GPU_TYPE -ARG FRAMEWORK - -ENV TASK_TYPE=${TASK_TYPE} -ENV GPU_TYPE=${GPU_TYPE} -ENV FRAMEWORK=${FRAMEWORK} ENV PYTHONUNBUFFERED=1 WORKDIR /app diff --git a/README.md b/README.md index 855532d..3b42c73 100644 --- a/README.md +++ b/README.md @@ -7,7 +7,7 @@ It satisfies the runtime contract from the parent README: - Root-level `Dockerfile` - Listens on port `8080` - Exposes `GET /health` -- Reads `TASK_TYPE`, `GPU_TYPE`, `FRAMEWORK`, and `EXTERNAL_SERVICE_TOKEN` +- Reads `EXTERNAL_SERVICE_TOKEN` - Handles `SIGTERM` for graceful shutdown ## Local Run @@ -21,11 +21,7 @@ curl http://localhost:8080/ ## Build Locally ```bash -docker build \ - --build-arg TASK_TYPE=validation \ - --build-arg GPU_TYPE=amd64 \ - --build-arg FRAMEWORK=pytorch \ - -t strategy-demo:v1.0.0 . +docker build -t strategy-demo:v1.0.0 . docker run --rm -p 8080:8080 strategy-demo:v1.0.0 ``` @@ -38,9 +34,6 @@ Push this directory to Gitea and keep the `v1.0.0` tag. Then create a strategy w { "name": "strategy-demo", "repo_url": "https:////strategy-demo", - "tag": "v1.0.0", - "task_type": "validation", - "gpu_type": "amd64", - "framework": "pytorch" + "tag": "v1.0.0" } ``` diff --git a/main.py b/main.py index ce54ed3..ae1b62f 100644 --- a/main.py +++ b/main.py @@ -12,9 +12,6 @@ shutdown_requested = False def _config() -> dict: token = os.getenv("EXTERNAL_SERVICE_TOKEN", "") return { - "task_type": os.getenv("TASK_TYPE", ""), - "gpu_type": os.getenv("GPU_TYPE", ""), - "framework": os.getenv("FRAMEWORK", ""), "external_service_token_present": bool(token), }