CI fixes

Author: crusaderky

.github/workflows/test-vendor.yml

@@ -46,6 +46,7 @@ jobs:
         uses: prefix-dev/[email protected]
         with:
           pixi-version: v0.39.0
+          manifest-path: array-api-extra/pyproject.toml
           cache: true
 
       - name: Test package

src/array_api_extra/_lib/_compat.py

@@ -3,9 +3,15 @@
 from __future__ import annotations
 
 try:
-    from ..._array_api_compat_vendor import array_namespace, device
+    from ..._array_api_compat_vendor import (  # pyright: ignore[reportMissingImports]
+        array_namespace,  # pyright: ignore[reportUnknownVariableType]
+        device,  # pyright: ignore[reportUnknownVariableType]
+    )
 except ImportError:
-    from array_api_compat import array_namespace, device
+    from array_api_compat import (  # pyright: ignore[reportMissingTypeStubs]
+        array_namespace,  # pyright: ignore[reportUnknownVariableType]
+        device,
+    )
 
 __all__ = [
     "array_namespace",

src/array_api_extra/_lib/_compat.pyi

@@ -0,0 +1,13 @@
+from types import ModuleType
+from ._typing import Array, Device
+
+class ArrayModule(ModuleType):
+    def device(self, x: Array, /) -> Device: ...
+
+def array_namespace(
+    *xs: Array, 
+    api_version: str | None = None,
+    use_compat: bool | None =None,
+) -> ArrayModule: ...
+
+def device(x: Array, /) -> Device: ...