Michael Rabinovich commited on
Commit
b1601ec
·
1 Parent(s): 380ae5d

dockerfile: track latest cadgenbench pre-v1

Browse files

Install cadgenbench from main so the eval image picks up the Open3D alignment dependency before the v1 release pin is set.

Files changed (2) hide show
  1. Dockerfile +5 -10
  2. README.md +3 -1
Dockerfile CHANGED
@@ -46,16 +46,11 @@ RUN python -m ensurepip --upgrade \
46
  && python -m pip install --no-cache-dir --upgrade pip
47
 
48
  # cadgenbench from the Public GitHub repo, same convention and ARG name
49
- # as the leaderboard Dockerfile. Defaults to `main` (every rebuild picks
50
- # up latest, pre-v1); lock to a commit SHA at the v1 release for
51
- # reproducible scores (see space-setup/post-gt-swap.md Stage F).
52
- ARG CADGENBENCH_SHA=289a2fc
53
- # Pinned to an explicit commit SHA, so the pip layer below is already
54
- # deterministic: changing the ARG changes the install command and busts
55
- # the cached layer on its own. (We previously cache-busted with an
56
- # `ADD https://api.github.com/.../commits/${SHA}`; GitHub's API now 403s
57
- # on the build IP, failing the build, and it was redundant for a fixed
58
- # SHA -- removed. Re-pin by bumping the ARG above, not via that ADD.)
59
  RUN python -m pip install --no-cache-dir \
60
  "cadgenbench @ git+https://github.com/huggingface/cadgenbench.git@${CADGENBENCH_SHA}"
61
 
 
46
  && python -m pip install --no-cache-dir --upgrade pip
47
 
48
  # cadgenbench from the Public GitHub repo, same convention and ARG name
49
+ # as the leaderboard Dockerfile. Defaults to `main` so every pre-v1 image
50
+ # rebuild picks up the latest evaluator dependencies (including Open3D for
51
+ # alignment). Lock to a commit SHA at the v1 release for reproducible scores
52
+ # (see space-setup/post-gt-swap.md Stage F).
53
+ ARG CADGENBENCH_SHA=main
 
 
 
 
 
54
  RUN python -m pip install --no-cache-dir \
55
  "cadgenbench @ git+https://github.com/huggingface/cadgenbench.git@${CADGENBENCH_SHA}"
56
 
README.md CHANGED
@@ -38,7 +38,9 @@ against this image on `a10g-large` and polls for completion.
38
  leaderboard Dockerfile) plus `libegl1 libegl-mesa0` for the EGL
39
  context.
40
  - `cadgenbench @ git+https://github.com/huggingface/cadgenbench@<sha>`,
41
- pinned via `ARG CADGENBENCH_SHA`.
 
 
42
  - `vtk-egl` swapped in for the PyPI `vtk` wheel (same swap shape as
43
  the leaderboard's `vtk-osmesa`; the GPU counterpart). PyVista
44
  picks up whichever `vtk` dist is installed; no cadgenbench code
 
38
  leaderboard Dockerfile) plus `libegl1 libegl-mesa0` for the EGL
39
  context.
40
  - `cadgenbench @ git+https://github.com/huggingface/cadgenbench@<sha>`,
41
+ controlled via `ARG CADGENBENCH_SHA` (pre-v1 defaults to `main`; pin a SHA
42
+ at release). This installs cadgenbench's Python dependencies, including
43
+ Open3D for rigid alignment.
44
  - `vtk-egl` swapped in for the PyPI `vtk` wheel (same swap shape as
45
  the leaderboard's `vtk-osmesa`; the GPU counterpart). PyVista
46
  picks up whichever `vtk` dist is installed; no cadgenbench code