Skip to content

Commit fc61d5c

Browse files
authored
fix: pin setuptools version to avoid compatibility issues (#1554)
1 parent fc912b0 commit fc61d5c

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

Dockerfile.tmpl

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,8 +27,9 @@ RUN uv pip install --no-build-isolation --no-cache --system "git+https://github.
2727
# b/404590350: Ray and torchtune have conflicting cli named `tune`. `ray` is not part of Colab's base image. Re-install `tune` to ensure the torchtune CLI is available by default.
2828
# b/468367647: Unpin protobuf, version greater than v5.29.5 causes issues with numerous packages
2929
# grpcio-tools must be installed here (not in kaggle_requirements.txt) to stay version-compatible with protobuf.
30+
# setuptools v82+ removed pkg_resources
3031
RUN uv pip install --system --force-reinstall --no-cache --no-deps torchtune
31-
RUN uv pip install --system --force-reinstall --no-cache "protobuf==5.29.5" "grpcio-tools>=1.60.0"
32+
RUN uv pip install --system --force-reinstall --no-cache "protobuf==5.29.5" "grpcio-tools>=1.60.0" "setuptools>=70.0,<82"
3233

3334
# Adding non-package dependencies:
3435
ADD clean-layer.sh /tmp/clean-layer.sh

0 commit comments

Comments
 (0)