@@ -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