diff --git a/pyproject.toml b/pyproject.toml index 0886e0380..31bac4b52 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -82,6 +82,7 @@ hf = [ "datasets>=3.0.0", "deepspeed>=0.9.6; platform_system != 'Darwin' and platform_system != 'Windows'", "diffusers>=0.32.2", + "httpx>=0.23.0,<1", "huggingface_hub>=0.24.0", "nltk", "peft>=0.17.0", diff --git a/tests/unit/torch/speculative/plugins/test_hf_streaming_dataset.py b/tests/unit/torch/speculative/plugins/test_hf_streaming_dataset.py index efd77267b..00b5a1f6a 100644 --- a/tests/unit/torch/speculative/plugins/test_hf_streaming_dataset.py +++ b/tests/unit/torch/speculative/plugins/test_hf_streaming_dataset.py @@ -30,12 +30,12 @@ import sys import types from unittest.mock import MagicMock -import httpx import pytest import torch # hf_streaming_dataset imports LabelSmoother at module scope. pytest.importorskip("transformers") +httpx = pytest.importorskip("httpx") from modelopt.torch.speculative.plugins import hf_streaming_dataset from modelopt.torch.speculative.plugins.hf_streaming_dataset import ( diff --git a/uv.lock b/uv.lock index 8a2d0efe1..bd9da2459 100644 --- a/uv.lock +++ b/uv.lock @@ -2620,6 +2620,7 @@ all = [ { name = "deepspeed", marker = "sys_platform != 'darwin' and sys_platform != 'win32'" }, { name = "diffusers" }, { name = "fire" }, + { name = "httpx" }, { name = "huggingface-hub" }, { name = "hydra-core" }, { name = "immutabledict" }, @@ -2658,6 +2659,7 @@ dev = [ { name = "deepspeed", marker = "sys_platform != 'darwin' and sys_platform != 'win32'" }, { name = "diffusers" }, { name = "fire" }, + { name = "httpx" }, { name = "huggingface-hub" }, { name = "hydra-core" }, { name = "immutabledict" }, @@ -2744,6 +2746,7 @@ hf = [ { name = "datasets" }, { name = "deepspeed", marker = "sys_platform != 'darwin' and sys_platform != 'win32'" }, { name = "diffusers" }, + { name = "httpx" }, { name = "huggingface-hub" }, { name = "nltk" }, { name = "peft" }, @@ -2794,6 +2797,7 @@ requires-dist = [ { name = "deepspeed", marker = "sys_platform != 'darwin' and sys_platform != 'win32' and extra == 'hf'", specifier = ">=0.9.6" }, { name = "diffusers", marker = "extra == 'hf'", specifier = ">=0.32.2" }, { name = "fire", marker = "extra == 'puzzletron'" }, + { name = "httpx", marker = "extra == 'hf'", specifier = ">=0.23.0,<1" }, { name = "huggingface-hub", marker = "extra == 'hf'", specifier = ">=0.24.0" }, { name = "hydra-core", marker = "extra == 'puzzletron'", specifier = "~=1.3.4" }, { name = "immutabledict", marker = "extra == 'puzzletron'" },