Merge pull request #13 from Parskatt/johan/pypi

add actions for pypi publishing and some simple tests

Author: Parskatt

.github/actions/uv-build/action.yml

@@ -0,0 +1,23 @@
+name: uv-build
+description: Build the project with uv and run smoke tests
+inputs:
+  python-version:
+    description: Python version to install
+    required: true
+runs:
+  using: composite
+  steps:
+    - name: Install uv
+      uses: astral-sh/setup-uv@v6
+    - name: Install Python
+      shell: bash
+      run: uv python install ${{ inputs.python-version }}
+    - name: Build
+      shell: bash
+      run: uv build
+    - name: Smoke test (wheel)
+      shell: bash
+      run: uv run --isolated --no-project --with dist/*.whl tests/smoke_test.py
+    - name: Smoke test (source distribution)
+      shell: bash
+      run: uv run --isolated --no-project --with dist/*.tar.gz tests/smoke_test.py

.github/workflows/build.yml

@@ -0,0 +1,22 @@
+name: Build
+
+on:
+  push:
+    branches:
+      - main
+  pull_request:
+    branches:
+      - main
+jobs:
+  run:
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: ['3.9', '3.10', '3.11', '3.12', '3.13']
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v5
+      - name: Build and smoke test
+        uses: ./.github/actions/uv-build
+        with:
+          python-version: ${{ matrix.python-version }}

.github/workflows/publish.yml

@@ -0,0 +1,39 @@
+name: PyPI Publish
+
+on:
+  push:
+    tags:
+      # Publish on any tag starting with a `v`, e.g., v0.1.0
+      - v*
+  workflow_dispatch:
+jobs:
+  build:
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: ['3.9', '3.10', '3.11', '3.12', '3.13']
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v5
+      - name: Build and smoke test
+        uses: ./.github/actions/uv-build
+        with:
+          python-version: ${{ matrix.python-version }}
+
+  publish:
+    runs-on: ubuntu-latest
+    needs: build
+    environment:
+      name: pypi
+    permissions:
+      id-token: write
+      contents: read
+    steps:
+      - name: Checkout
+        uses: actions/checkout@v5
+      - name: Build and smoke test
+        uses: ./.github/actions/uv-build
+        with:
+          python-version: '3.12'
+      - name: Publish
+        run: uv publish

pyproject.toml

@@ -23,12 +23,18 @@ dev = [
     "ruff>=0.9.10",
     "tqdm>=4.67.1",
     "wandb>=0.19.8",
+    "pytest>=8.4.2",
 ]
 
 
 [build-system]
-requires = ["hatchling"]
-build-backend = "hatchling.build"
+requires = ["uv_build>=0.8.14,<0.9.0"]
+build-backend = "uv_build"
+
+
+[tool.uv.build-backend]
+module-name = "dad"
+module-root = ""
 
 
 [tool.uv.sources]

tests/smoke_test.py

@@ -0,0 +1,6 @@
+import dad
+
+
+def test_smoke():
+    model = dad.load_DaD()
+    assert model is not None

tests/vis_test.py

@@ -0,0 +1,9 @@
+import dad
+from dad.utils import visualize_keypoints
+
+
+def test_vis():
+    detector = dad.load_DaD()
+    img_path = "assets/0015_A.jpg"
+    vis_path = "vis/0015_A_dad.jpg"
+    visualize_keypoints(img_path, vis_path, detector, num_keypoints = 512)