Fix CI (inference tests, dataset viewer user, mypy) (#2208)

* add cassettes for InferenceAPI tests * Fix dataset viewer user * fix mypy * set UV_HTTP_TIMEOUT
huggingface · Apr 9, 2024 · dc824a5 · dc824a5
1 parent b2a8768
commit dc824a5
Show file tree

Hide file tree

Showing 13 changed files with 863 additions and 4 deletions.
diff --git a/.github/workflows/python-quality.yml b/.github/workflows/python-quality.yml
@@ -16,6 +16,8 @@ on:
 jobs:
   check_code_quality:
     runs-on: ubuntu-latest
+    env:
+      UV_HTTP_TIMEOUT: 600 # max 10min to install deps
 
     steps:
       - uses: actions/checkout@v2

diff --git a/.github/workflows/python-tests.yml b/.github/workflows/python-tests.yml
@@ -15,6 +15,9 @@ on:
 jobs:
   build-ubuntu:
     runs-on: ubuntu-latest
+    env:
+      UV_HTTP_TIMEOUT: 600 # max 10min to install deps
+
     strategy:
       fail-fast: false
       matrix:
@@ -139,6 +142,7 @@ jobs:
     runs-on: windows-latest
     env:
       DISABLE_SYMLINKS_IN_WINDOWS_TESTS: 1
+      UV_HTTP_TIMEOUT: 600 # max 10min to install deps
     strategy:
       fail-fast: false
       matrix:

diff --git a/src/huggingface_hub/inference/_client.py b/src/huggingface_hub/inference/_client.py
@@ -114,7 +114,7 @@
 
 if TYPE_CHECKING:
     import numpy as np
-    from PIL import Image
+    from PIL.Image import Image
 
 logger = logging.getLogger(__name__)
 

diff --git a/src/huggingface_hub/inference/_common.py b/src/huggingface_hub/inference/_common.py
@@ -70,7 +70,7 @@
 
 if TYPE_CHECKING:
     from aiohttp import ClientResponse, ClientSession
-    from PIL import Image
+    from PIL.Image import Image
 
 # TYPES
 UrlT = str

diff --git a/src/huggingface_hub/inference/_generated/_async_client.py b/src/huggingface_hub/inference/_generated/_async_client.py
@@ -99,7 +99,7 @@
 
 if TYPE_CHECKING:
     import numpy as np
-    from PIL import Image
+    from PIL.Image import Image
 
 logger = logging.getLogger(__name__)
 

diff --git a/tests/cassettes/InferenceApiTest.test_inference_missing_input.yaml b/tests/cassettes/InferenceApiTest.test_inference_missing_input.yaml
diff --git a/tests/cassettes/InferenceApiTest.test_inference_overriding_invalid_task.yaml b/tests/cassettes/InferenceApiTest.test_inference_overriding_invalid_task.yaml
diff --git a/tests/cassettes/InferenceApiTest.test_inference_overriding_task.yaml b/tests/cassettes/InferenceApiTest.test_inference_overriding_task.yaml
diff --git a/tests/cassettes/InferenceApiTest.test_simple_inference.yaml b/tests/cassettes/InferenceApiTest.test_simple_inference.yaml
diff --git a/tests/cassettes/InferenceApiTest.test_text_to_image.yaml b/tests/cassettes/InferenceApiTest.test_text_to_image.yaml
diff --git a/tests/cassettes/InferenceApiTest.test_text_to_image_raw_response.yaml b/tests/cassettes/InferenceApiTest.test_text_to_image_raw_response.yaml
diff --git a/tests/test_inference_api.py b/tests/test_inference_api.py
@@ -15,6 +15,7 @@
 from pathlib import Path
 from unittest.mock import patch
 
+import pytest
 from PIL import Image
 
 from huggingface_hub import hf_hub_download
@@ -23,6 +24,7 @@
 from .testing_utils import expect_deprecation, with_production_testing
 
 
+@pytest.mark.vcr
 @with_production_testing
 class InferenceApiTest(unittest.TestCase):
     def read(self, filename: str) -> bytes:

diff --git a/tests/testing_constants.py b/tests/testing_constants.py
@@ -7,7 +7,7 @@
 
 # Used to create repos that we don't own (example: for gated repo)
 # Token is not critical. Also public in https://github.com/huggingface/datasets-server
-OTHER_USER = "DSSUser"
+OTHER_USER = "DVUser"
 OTHER_TOKEN = "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"
 
 ENDPOINT_PRODUCTION = "https://huggingface.co"