diff --git a/pyproject.toml b/pyproject.toml index 30a31d46f..5fd603fe4 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -9,7 +9,7 @@ requires = [ "torch == 2.9.1", "wheel", "jinja2", - "grpcio-tools", + "grpcio-tools==1.78.0", ] build-backend = "setuptools.build_meta" diff --git a/requirements/build.txt b/requirements/build.txt index 0d3a24dd1..588a709c9 100644 --- a/requirements/build.txt +++ b/requirements/build.txt @@ -10,4 +10,4 @@ jinja2>=3.1.6 regex build protobuf -grpcio-tools +grpcio-tools==1.78.0 # Required for grpc entrypoints diff --git a/requirements/rocm.txt b/requirements/rocm.txt index dc407995a..7ac853680 100644 --- a/requirements/rocm.txt +++ b/requirements/rocm.txt @@ -15,4 +15,4 @@ setuptools-scm>=8 runai-model-streamer[s3,gcs]==0.15.3 conch-triton-kernels==1.2.1 timm>=1.0.17 -grpcio-tools>=1.76.0 \ No newline at end of file +grpcio-tools==1.78.0 # Should match `build.txt` \ No newline at end of file diff --git a/requirements/test.in b/requirements/test.in index 3f2110f82..ea4a07784 100644 --- a/requirements/test.in +++ b/requirements/test.in @@ -48,6 +48,7 @@ buildkite-test-collector==0.1.9 genai_perf>=0.0.8 tritonclient>=2.51.0 +grpcio-tools==1.78.0 # Should match `build.txt` arctic-inference == 0.1.1 # Required for suffix decoding test numba == 0.61.2 # Required for N-gram speculative decoding numpy diff --git a/requirements/test.txt b/requirements/test.txt index af6a276c2..2d49db348 100644 --- a/requirements/test.txt +++ b/requirements/test.txt @@ -303,8 +303,12 @@ graphql-relay==3.2.0 # via graphene greenlet==3.2.3 # via sqlalchemy -grpcio==1.76.0 - # via ray +grpcio==1.78.0 + # via + # grpcio-tools + # ray +grpcio-tools==1.78.0 + # via -r requirements/test.in gunicorn==23.0.0 # via mlflow h11==0.14.0 @@ -777,6 +781,7 @@ protobuf==6.33.2 # via # google-api-core # googleapis-common-protos + # grpcio-tools # mlflow-skinny # opentelemetry-proto # proto-plus @@ -1046,6 +1051,7 @@ sentence-transformers==5.2.0 # mteb setuptools==77.0.3 # via + # grpcio-tools # lightning-utilities # pytablewriter # torch