Skip to content

Commit 210bce9

Browse files
committed
fix version checks
1 parent 064cc7d commit 210bce9

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

tests/integrations/huggingface_hub/test_huggingface_hub.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -108,7 +108,7 @@ def mock_hf_text_generation_api(httpx_mock):
108108
status=200,
109109
)
110110

111-
if HF_VERSION > (1, 0, 0):
111+
if HF_VERSION >= (1, 0, 0):
112112
yield httpx_mock
113113
else:
114114
yield rsps
@@ -160,7 +160,7 @@ def mock_hf_api_with_errors(httpx_mock):
160160
status=500,
161161
)
162162

163-
if HF_VERSION > (1, 0, 0):
163+
if HF_VERSION >= (1, 0, 0):
164164
yield httpx_mock
165165
else:
166166
yield rsps
@@ -210,7 +210,7 @@ def mock_hf_text_generation_api_streaming(httpx_mock):
210210
},
211211
)
212212

213-
if HF_VERSION > (1, 0, 0):
213+
if HF_VERSION >= (1, 0, 0):
214214
yield httpx_mock
215215
else:
216216
yield rsps
@@ -273,7 +273,7 @@ def mock_hf_chat_completion_api(httpx_mock):
273273
status=200,
274274
)
275275

276-
if HF_VERSION > (1, 0, 0):
276+
if HF_VERSION >= (1, 0, 0):
277277
yield httpx_mock
278278
else:
279279
yield rsps
@@ -345,7 +345,7 @@ def mock_hf_chat_completion_api_tools(httpx_mock):
345345
status=200,
346346
)
347347

348-
if HF_VERSION > (1, 0, 0):
348+
if HF_VERSION >= (1, 0, 0):
349349
yield httpx_mock
350350
else:
351351
yield rsps
@@ -398,7 +398,7 @@ def mock_hf_chat_completion_api_streaming(httpx_mock):
398398
},
399399
)
400400

401-
if HF_VERSION > (1, 0, 0):
401+
if HF_VERSION >= (1, 0, 0):
402402
yield httpx_mock
403403
else:
404404
yield rsps
@@ -451,7 +451,7 @@ def mock_hf_chat_completion_api_streaming_tools(httpx_mock):
451451
},
452452
)
453453

454-
if HF_VERSION > (1, 0, 0):
454+
if HF_VERSION >= (1, 0, 0):
455455
yield httpx_mock
456456
else:
457457
yield rsps

0 commit comments

Comments
 (0)