build(deps): bump nvidia-modelopt to 0.43

.github/workflows/_build_test_publish_wheel.yml

@@ -74,7 +74,7 @@ jobs:
             rm LICENSE || true
             docker run --rm -e NO_VCS_VERSION=1 -v $(pwd):/workspace -w /workspace $IMAGE bash -c '\
               for python_version in cp311 cp312 cp313; do \
-                /opt/python/${python_version}-${python_version}/bin/pip install --upgrade "setuptools<80.0.0,>=77.0.0" build; \
+                /opt/python/${python_version}-${python_version}/bin/pip install --upgrade "setuptools>=80" build; \
               done && \
               for python_version in cp311 cp312 cp313; do \
                 /opt/python/${python_version}-${python_version}/bin/python -m build; \

docker/Dockerfile.ci.dev

@@ -2,7 +2,7 @@
 # syntax=docker/dockerfile:1.3-labs
 
 ARG FROM_IMAGE_NAME
 FROM ${FROM_IMAGE_NAME} as main
 ENV PIP_CONSTRAINT=""
 ENV DEBIAN_FRONTEND=noninteractive
 ARG UV_VERSION=0.7.2
@@ -82,7 +82,7 @@
 ENV UV_PYTHON=$UV_PROJECT_ENVIRONMENT/bin/python
 
 ##### For NVIDIANS only #####
 FROM main as jet
 ARG JET_API_VERSION
 ENV PATH="$PATH:/opt/jet/bin"
 RUN --mount=type=secret,id=JET_INDEX_URLS bash -ex <<"EOF"
@@ -97,7 +97,7 @@
     JET_INDEX_URLS=$(cat /run/secrets/JET_INDEX_URLS)
     LOGGER_INDEX_URL=$(cat /run/secrets/LOGGER_INDEX_URL)
     uv pip install --no-cache-dir --upgrade $LOGGER_INDEX_URL "one-logger"
-    uv pip install --no-cache-dir --upgrade "setuptools<80.0.0,>=77.0.0"
+    uv pip install --no-cache-dir --upgrade "setuptools>=80"
     uv pip install --no-cache-dir --upgrade $JET_INDEX_URLS "jet-client~=4.0" 
 EOF
 ###

docker/common/install.sh

@@ -136,7 +136,7 @@ main() {
         . $UV_PROJECT_ENVIRONMENT/bin/activate
 
         pip install --pre --no-cache-dir --upgrade pip
-        pip install --pre --no-cache-dir torch pybind11 wheel_stub ninja wheel packaging "setuptools<80.0.0,>=77.0.0"
+        pip install --pre --no-cache-dir torch pybind11 wheel_stub ninja wheel packaging "setuptools>=80"
         pip install --pre --no-cache-dir --no-build-isolation .
     fi
 

docker/common/install_source_wheels.sh

@@ -50,4 +50,4 @@ fi
 uv pip install --no-cache-dir \
     $MAMBA_WHEEL \
     $CAUSALCONV1D_WHEEL \
-    "setuptools<80.0.0,>=77.0.0"
+    "setuptools>=80"

megatron/core/distributed/fsdp/src/pyproject.toml

@@ -1,7 +1,7 @@
 # Copyright (c) 2023, NVIDIA CORPORATION.  All rights reserved.
 
 [build-system]
-requires = ["setuptools<80.0.0", "pybind11"]
+requires = ["setuptools>=80", "pybind11"]
 build-backend = "setuptools.build_meta"
 
 [tool.setuptools]

pyproject.toml

@@ -1,7 +1,7 @@
 # Copyright (c) 2023, NVIDIA CORPORATION.  All rights reserved.
 
 [build-system]
-requires = ["setuptools<80.0.0", "pybind11", "packaging>=24.2"]
+requires = ["setuptools>=80", "pybind11", "packaging>=24.2"]
 build-backend = "setuptools.build_meta"
 
 [tool.setuptools]
@@ -88,7 +88,7 @@ mlm = [
 ]
 
 dev = [
-    "nvidia-modelopt[torch]; sys_platform != 'darwin'",
+    "nvidia-modelopt[torch]>=0.43; sys_platform != 'darwin'",
     "nvidia-resiliency-ext",
     "tqdm",
     "einops~=0.8",
@@ -161,7 +161,7 @@ docs = [
     "nvidia-sphinx-theme", # Our NVIDIA theme
 ]
 build = [
-    "setuptools<80.0.0,>=77.0.0",
+    "setuptools>=80",
     "packaging>=24.2",
     "hatchling",
     "pybind11",