fix skipping condition

huggingface · Feb 13, 2025 · c478fe6 · c478fe6
1 parent d8f5415
commit c478fe6
Show file tree

Hide file tree

Showing 7 changed files with 4,339 additions and 1,076 deletions.
diff --git a/....test_feature_extraction_with_sentence_transformers[hf-inference,feature-extraction].yaml b/....test_feature_extraction_with_sentence_transformers[hf-inference,feature-extraction].yaml
diff --git a/...nceClient.test_feature_extraction_with_transformers[hf-inference,feature-extraction].yaml b/...nceClient.test_feature_extraction_with_transformers[hf-inference,feature-extraction].yaml
diff --git a/...ssettes/TestInferenceClient.test_image_segmentation[hf-inference,image-segmentation].yaml b/...ssettes/TestInferenceClient.test_image_segmentation[hf-inference,image-segmentation].yaml
diff --git a/...ettes/TestInferenceClient.test_text_classification[hf-inference,text-classification].yaml b/...ettes/TestInferenceClient.test_text_classification[hf-inference,text-classification].yaml
@@ -13,78 +13,49 @@ interactions:
       Content-Type:
       - application/json
       X-Amzn-Trace-Id:
-      - 31b9171d-7054-49d0-b24c-9038ca30094e
-    method: POST
-    uri: https://router.huggingface.co/hf-inference/models/distilbert/distilbert-base-uncased-finetuned-sst-2-english
-  response:
-    body:
-      string: '{"error":"Model distilbert/distilbert-base-uncased-finetuned-sst-2-english
-        is currently loading","estimated_time":20.0}'
-    headers:
-      Connection:
-      - keep-alive
-      Content-Length:
-      - '119'
-      Content-Type:
-      - application/json
-      Date:
-      - Wed, 22 Jan 2025 14:12:36 GMT
-      access-control-allow-credentials:
-      - 'true'
-      vary:
-      - Origin, Access-Control-Request-Method, Access-Control-Request-Headers
-      x-request-id:
-      - NBkRpeOxOFqqVgoQbQBkR
-    status:
-      code: 503
-      message: Service Unavailable
-- request:
-    body: '{"inputs": "I like you", "parameters": {}}'
-    headers:
-      Accept:
-      - '*/*'
-      Accept-Encoding:
-      - gzip, deflate
-      Connection:
-      - keep-alive
-      Content-Length:
-      - '42'
-      Content-Type:
-      - application/json
-      X-Amzn-Trace-Id:
-      - b50dcc59-497a-4c68-98c2-99570aa03d0e
-      X-wait-for-model:
-      - '1'
+      - 89809c7c-fa69-4600-b32b-dbf6a84f19d2
     method: POST
     uri: https://router.huggingface.co/hf-inference/models/distilbert/distilbert-base-uncased-finetuned-sst-2-english
   response:
     body:
       string: '[[{"label":"POSITIVE","score":0.9998695850372314},{"label":"NEGATIVE","score":0.00013043530634604394}]]'
     headers:
+      Access-Control-Allow-Origin:
+      - '*'
+      Access-Control-Expose-Headers:
+      - X-Repo-Commit,X-Request-Id,X-Error-Code,X-Error-Message,X-Total-Count,ETag,Link,Accept-Ranges,Content-Range,X-Xet-Access-Token,X-Xet-Token-Expiration,X-Xet-Refresh-Route,X-Xet-Cas-Url,X-Xet-Hash
       Connection:
       - keep-alive
+      Content-Length:
+      - '103'
       Content-Type:
       - application/json
       Date:
-      - Wed, 22 Jan 2025 14:12:46 GMT
-      Transfer-Encoding:
-      - chunked
+      - Thu, 13 Feb 2025 18:51:17 GMT
+      Referrer-Policy:
+      - strict-origin-when-cross-origin
+      Via:
+      - 1.1 10e1486bad24a11a3edd4fce423c3662.cloudfront.net (CloudFront)
+      X-Amz-Cf-Id:
+      - LIGzQVr8RIM0oAYAamVg6C2c0wOgJY6mtYCPOKhocStmxeoR6IOajg==
+      X-Amz-Cf-Pop:
+      - CDG55-P3
+      X-Cache:
+      - Miss from cloudfront
+      X-Powered-By:
+      - huggingface-moon
       access-control-allow-credentials:
       - 'true'
-      access-control-expose-headers:
-      - x-compute-type, x-compute-time
-      server:
-      - uvicorn
+      cross-origin-opener-policy:
+      - same-origin
       vary:
       - Origin, Access-Control-Request-Method, Access-Control-Request-Headers
-      x-compute-characters:
-      - '10'
       x-compute-time:
-      - '0.021'
+      - '0.024'
       x-compute-type:
-      - cpu
+      - cache
       x-request-id:
-      - EGClKTkUnxpsN3CQdnvKM
+      - Root=1-67ae3f25-66173bfc00b9f87e11864513;89809c7c-fa69-4600-b32b-dbf6a84f19d2
       x-sha:
       - 714eb0fa89d2f80546fda750413ed43d93601a13
     status:

diff --git a/tests/cassettes/TestInferenceClient.test_text_to_speech[hf-inference,text-to-speech].yaml b/tests/cassettes/TestInferenceClient.test_text_to_speech[hf-inference,text-to-speech].yaml
diff --git a/...tes/TestInferenceClient.test_token_classification[hf-inference,token-classification].yaml b/...tes/TestInferenceClient.test_token_classification[hf-inference,token-classification].yaml
@@ -1,44 +1,4 @@
 interactions:
-- request:
-    body: '{"inputs": "My name is Sarah Jessica Parker but you can call me Jessica",
-      "parameters": {}}'
-    headers:
-      Accept:
-      - '*/*'
-      Accept-Encoding:
-      - gzip, deflate
-      Connection:
-      - keep-alive
-      Content-Length:
-      - '91'
-      Content-Type:
-      - application/json
-      X-Amzn-Trace-Id:
-      - bc8215b9-5548-4313-a0ad-4acacc909054
-    method: POST
-    uri: https://router.huggingface.co/hf-inference/models/dbmdz/bert-large-cased-finetuned-conll03-english
-  response:
-    body:
-      string: '{"error":"Model dbmdz/bert-large-cased-finetuned-conll03-english is
-        currently loading","estimated_time":53.3741569519043}'
-    headers:
-      Connection:
-      - keep-alive
-      Content-Length:
-      - '121'
-      Content-Type:
-      - application/json
-      Date:
-      - Wed, 22 Jan 2025 14:13:35 GMT
-      access-control-allow-credentials:
-      - 'true'
-      vary:
-      - Origin, Access-Control-Request-Method, Access-Control-Request-Headers
-      x-request-id:
-      - 1U3Dh107VU9iZkoMFaIyJ
-    status:
-      code: 503
-      message: Service Unavailable
 - request:
     body: '{"inputs": "My name is Sarah Jessica Parker but you can call me Jessica",
       "parameters": {}}'

diff --git a/tests/test_inference_client.py b/tests/test_inference_client.py
@@ -219,9 +219,10 @@ def client(request):
     """
     provider, model, api_key = request.param
     vcr_record_mode = request.config.getoption("--vcr-record")
-
     # If we are recording and the api key is not set, skip the test
-    if vcr_record_mode != "none" and not api_key:
+    # replaying modes are "all", "new_episodes" and "once"
+    # non replaying modes are "none" and None
+    if vcr_record_mode not in ["none", None] and not api_key:
         pytest.skip(f"API KEY not set for provider {provider}, skipping test")
 
     # If api_key is provided, use it