Backport PR #1596: GPU tests: install with uv (#1618)

Author: ilan-gold

.github/workflows/test-gpu.yml

@@ -51,23 +51,25 @@ jobs:
       - name: Nvidia SMI sanity check
         run: nvidia-smi
 
-      - uses: mamba-org/setup-micromamba@v1
+      - name: Install Python
+        uses: actions/setup-python@v5
         with:
-          micromamba-version: "1.5.6-0"
-          environment-file: ci/gpu_ci.yml
-          post-cleanup: "all"
+          python-version: "3.x"
+
+      - name: Install UV
+        uses: hynek/setup-cached-uv@v2
+        with:
+          cache-dependency-path: pyproject.toml
 
       - name: Install AnnData
-        run: uv pip install -e .[dev,test,gpu]
+        run: uv pip install --system -e ".[dev,test,cu12]" llvmlite>=0.43
 
       - name: Env list
-        run: |
-          micromamba list
-          pip list
+        run: pip list
 
       - name: Run test
         run: pytest -m gpu --cov --cov-report=xml --cov-context=test -n 4
 
-      - uses: codecov/codecov-action@v3
+      - uses: codecov/codecov-action@v4
         with:
           flags: gpu-tests

ci/gpu_ci.yml

@@ -0,0 +1 @@
+create new `cupy` installation options for cuda 11 & 12 called `cu11` and `cu12` {user}`Intron7`

docs/release-notes/1596.dev.md

@@ -101,10 +101,9 @@ test = [
     "pytest_memray",
     "pytest-mock"
 ]
-gpu = [
-    "cupy",
-    "numpy<2.0.0",
-]
+gpu = ["cupy"]
+cu12 = ["cupy-cuda12x"]
+cu11 = ["cupy-cuda11x"]
 
 [tool.hatch.version]
 source = "vcs"
@@ -212,3 +211,6 @@ title_format = "(v{version})=\n### {version} {{small}}`{project_date}`"
 
 [tool.towncrier.fragment.performance]
 name = "Performance"
+
+[tool.towncrier.fragment.dev]
+name = "Development Process"