Update Dockerfile

Author: nguu0123

pyproject.toml

@@ -5,7 +5,7 @@ description = "Add your description here"
 authors = [{ name = "nguu0123", email = "nguu0123@gmail.com" }]
 dependencies = []
 readme = "README.md"
-requires-python = ">= 3.10"
+requires-python = "== 3.10.18"
 
 [tool.uv.sources]
 util = { workspace = true }

src/ensemble/Dockerfile

@@ -1,27 +1,25 @@
+# This's run from root dir
 FROM debian:stable-slim
 
-WORKDIR /ensemble
+WORKDIR /workspace
 
 COPY --from=ghcr.io/astral-sh/uv:0.9.0 /uv /uvx /bin/
 
-COPY ./ensemble/*.toml ./
+COPY pyproject.toml uv.lock /workspace/
 
-COPY ./ensemble/run_server.sh ./ensemble/run_server.sh
+RUN uv sync --frozen --no-install-workspace --package=ensemble --no-cache --compile-bytecode
 
-RUN uv python pin 3.10.18
+COPY src/ensemble /workspace/src/ensemble
 
-RUN uv sync --compile-bytecode --no-cache
+COPY src/util /workspace/src/util
 
-COPY ./ensemble/*.py ./ensemble/
-
-COPY ./ensemble/*.yaml ./ensemble/
-
-COPY ./util/ ./util
+RUN uv sync --frozen --package=ensemble --no-cache --compile-bytecode
 
 EXPOSE 5011
 
-WORKDIR /ensemble/ensemble
+ENV PATH="/workspace/.venv/bin:$PATH"
 
-ENV PATH="/ensemble/.venv/bin:$PATH"
+WORKDIR /workspace/src/ensemble
 
 ENTRYPOINT [ "./run_server.sh" ]
+

src/ensemble/docker_build.sh

@@ -11,8 +11,12 @@ dependencies = [
   "opentelemetry-instrumentation-aiohttp-client>=0.48b0",
   "aio-pika>=9.5.5",
   "opentelemetry-instrumentation-aio-pika>=0.54b1",
+  "util",
 ]
 
 name = "ensemble"
 version = "0.1.0"
 requires-python = "== 3.10.18"
+
+[tool.uv.sources]
+util = { workspace = true }

src/ensemble/docker_run.sh

@@ -1,29 +1,24 @@
 FROM rdsea/debian-slim-opencv:latest
 
-WORKDIR /object_classification
-
-COPY ./inference/onnx_model/ ./inference/onnx_model/
+WORKDIR /workspace
 
 COPY --from=ghcr.io/astral-sh/uv:0.9.0 /uv /uvx /bin/
 
-COPY ./inference/*.toml ./
-
-RUN uv python pin 3.10.18
+COPY pyproject.toml uv.lock /workspace/
 
-RUN uv sync --compile-bytecode --no-cache --extra cpu
+RUN uv sync --frozen --no-install-workspace --package=inference --no-cache --compile-bytecode
 
-COPY ./inference/*.py ./inference/
+COPY src/inference /workspace/src/inference
 
-COPY ./inference/*.yaml ./inference/
+COPY src/util /workspace/src/util
 
-COPY ./util/ ./util
-
-COPY ./inference/run_server.sh ./inference/run_server.sh
+RUN uv sync --frozen --package=inference --no-cache --compile-bytecode
 
 EXPOSE 5012
 
-WORKDIR /object_classification/inference
+ENV PATH="/workspace/.venv/bin:$PATH"
 
-ENV PATH="/object_classification/.venv/bin:$PATH"
+WORKDIR /workspace/src/inference
 
 ENTRYPOINT [ "./run_server.sh" ]
+

src/ensemble/pyproject.toml

@@ -8,30 +8,18 @@ RUN apt-get install -y ffmpeg libsm6 libxext6 wget
 
 COPY --from=ghcr.io/astral-sh/uv:0.9.0 /uv /uvx /bin/
 
-COPY ./inference/onnx_model/ ./inference/onnx_model/
+COPY src/inference /workspace/src/inference
 
-COPY ./inference/run_server.sh ./inference/run_server.sh
+COPY src/util /workspace/src/util
 
-COPY ./inference/*.toml ./
-
-RUN uv python pin 3.10.18
-
-RUN uv sync --compile-bytecode --no-cache
+RUN uv sync --frozen --package=inference --no-cache --compile-bytecode
 
 RUN wget https://pypi.jetson-ai-lab.io/jp6/cu126/+f/e1e/9e3dc2f4d5551/onnxruntime_gpu-1.23.0-cp310-cp310-linux_aarch64.whl#sha256=e1e9e3dc2f4d5551c5f0b5554cf490d141cd0d339a5a7f4826ac0e04f20a35fc
 
-RUN uv pip install ./onnxruntime_gpu-1.23.0-cp310-cp310-linux_aarch64.whl
-
-COPY ./inference/*.py ./inference/
-
-COPY ./inference/*.yaml ./inference/
-
-COPY ./util/ ./util
-
-EXPOSE 8050-8065
+RUN uv pip install ./onnxruntime_gpu-1.23.0-cp310-cp310-linux_aarch64.whl && rm ./onnxruntime_gpu-1.23.0-cp310-cp310-linux_aarch64.whl
 
-WORKDIR /object_classification/inference
+WORKDIR /workspace/src/inference
 
-ENV PATH="/object_classification/.venv/bin:$PATH"
+ENV PATH="/workspace/.venv/bin:$PATH"
 
 ENTRYPOINT [ "./run_server.sh" ]

src/inference/Dockerfile.cpu

@@ -10,6 +10,7 @@ dependencies = [
   "opentelemetry-exporter-otlp-proto-http>=1.27.0",
   "opentelemetry-exporter-otlp>=1.27.0",
   "opentelemetry-instrumentation-aiohttp-client>=0.48b0",
+  "util",
 ]
 
 optional-dependencies.cpu = ["onnxruntime"]

src/inference/Dockerfile.gpu

@@ -1,30 +1,25 @@
+# This's run from root dir
 FROM rdsea/debian-slim-opencv:latest
 
-WORKDIR /image_preprocessing/preprocessing
+WORKDIR /workspace
 
 COPY --from=ghcr.io/astral-sh/uv:0.9.0 /uv /uvx /bin/
 
-COPY ./preprocessing/*.toml ./
+COPY pyproject.toml uv.lock /workspace/
 
-RUN uv python pin 3.10.18
+RUN uv sync --frozen --no-install-workspace --package=preprocessing --no-cache --compile-bytecode
 
-RUN uv sync --compile-bytecode --no-cache
+COPY src/preprocessing /workspace/src/preprocessing
 
-WORKDIR /image_preprocessing
+COPY src/util /workspace/src/util
 
-COPY ./preprocessing/run_server.sh ./preprocessing/run_server.sh
-
-COPY ./preprocessing/*.yaml ./preprocessing/
-
-COPY ./preprocessing/*.py ./preprocessing/
-
-COPY ./util/ ./util
+RUN uv sync --frozen --package=preprocessing --no-cache --compile-bytecode
 
 EXPOSE 5010
 
-WORKDIR /image_preprocessing/preprocessing
+ENV PATH="/workspace/.venv/bin:$PATH"
 
-ENV PATH="/image_preprocessing/preprocessing/.venv/bin:$PATH"
+WORKDIR /workspace/src/preprocessing
 
 ENTRYPOINT [ "./run_server.sh" ]