Skip to content

Commit 34501c9

Browse files
committed
fix test
1 parent 061b8ef commit 34501c9

File tree

2 files changed

+5
-0
lines changed

2 files changed

+5
-0
lines changed

pyproject.toml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@ dependencies = [
2323
[project.optional-dependencies]
2424
dev = [
2525
"pytest>=7.0.0",
26+
"pytest-asyncio>=0.21.0",
2627
"black>=22.0.0",
2728
"isort>=5.10.0",
2829
"mypy>=0.950",

tests/test_utils.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,10 @@ def start_test_server(model: str = TEST_MODEL, port: Optional[int] = None) -> Tu
5353
env = os.environ.copy()
5454
env["OPTILLM_API_KEY"] = "optillm"
5555

56+
# Pass HF_TOKEN if available (needed for model downloads in CI)
57+
if "HF_TOKEN" in os.environ:
58+
env["HF_TOKEN"] = os.environ["HF_TOKEN"]
59+
5660
print(f"Starting optillm server on port {port}...")
5761

5862
# Start server (don't capture output to avoid pipe buffer deadlock)

0 commit comments

Comments
 (0)