chore: migrate from pylint to ruff, update CI workflows

- Replace pylint with ruff for linting and formatting
- Update CI workflows to use uv and correct package name
- Add release workflow for PyPI publishing
- Add requires-python constraint
- Fix code style issues (imports, quotes, comparisons)
- Add initial test for get_mean_appearance

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

Author: AmitMY

.github/workflows/lint.yaml

@@ -3,24 +3,31 @@ name: Lint
 
 on:
   push:
-    branches: [ master, main ]
+    branches: [ main ]
   pull_request:
-    branches: [ master, main ]
+    branches: [ main ]
 
 
 jobs:
-  test:
+  lint:
     name: Lint
     runs-on: ubuntu-latest
 
     steps:
-      - uses: actions/checkout@v3
-      - uses: actions/setup-python@v4
+      - uses: actions/checkout@v6
+
+      - name: Setup uv
+        uses: astral-sh/setup-uv@v7
         with:
-          python-version: '3.10'
+          python-version: "3.9"
+          enable-cache: true
+          activate-environment: true
+
+      - name: Install dependencies
+        run: uv pip install ".[dev]"
 
-      - name: Install Requirements
-        run: pip install .[dev]
+      - name: Lint code
+        run: uv run ruff check pose_anonymization
 
-      - name: Lint Code
-        run: pylint pose_anonymization
+      - name: Check formatting
+        run: uv run ruff format --check pose_anonymization

.github/workflows/release.yaml

@@ -0,0 +1,39 @@
+name: Publish Python Package
+on:
+  release:
+    types: [ created ]
+
+jobs:
+  pypi-publish:
+    name: Upload release to PyPI
+    runs-on: ubuntu-latest
+    environment:
+      name: pypi
+      url: https://pypi.org/p/pose-anonymization
+    permissions:
+      id-token: write
+    steps:
+      - uses: actions/checkout@v5
+
+      - uses: actions/setup-python@v6
+        with:
+          python-version: "3.12"
+
+      - name: Extract release version
+        id: get_version
+        run: echo "version=${GITHUB_REF#refs/tags/}" >> $GITHUB_ENV
+
+      - name: Update version in pyproject.toml
+        run: |
+          sed -i 's/^version = .*/version = "${{ env.version }}"/' pyproject.toml
+
+      - name: Install build dependencies
+        run: pip install build
+
+      - name: Build a binary wheel dist
+        run: |
+          rm -rf dist
+          python -m build
+
+      - name: Publish distribution 📦 to PyPI
+        uses: pypa/gh-action-pypi-publish@release/v1

.github/workflows/test.yaml

@@ -3,9 +3,9 @@ name: Test
 
 on:
   push:
-    branches: [ master, main ]
+    branches: [ main ]
   pull_request:
-    branches: [ master, main ]
+    branches: [ main ]
 
 
 jobs:
@@ -14,13 +14,17 @@ jobs:
     runs-on: ubuntu-latest
 
     steps:
-      - uses: actions/checkout@v3
-      - uses: actions/setup-python@v4
+      - uses: actions/checkout@v6
+
+      - name: Setup uv
+        uses: astral-sh/setup-uv@v7
         with:
-          python-version: '3.10'
+          python-version: "3.9"
+          enable-cache: true
+          activate-environment: true
 
-      - name: Install Requirements
-        run: pip install .[dev]
+      - name: Install dependencies
+        run: uv pip install ".[dev]"
 
       - name: Test Code
-        run: pytest pose_anonymization
+        run: uv run pytest

pose_anonymization/appearance.py

@@ -3,10 +3,9 @@
 import numpy as np
 from pose_format import Pose
 from pose_format.numpy import NumPyPoseBody
-from pose_format.utils.generic import pose_normalization_info
+from pose_format.utils.generic import normalize_pose_size, pose_normalization_info
 
 from pose_anonymization.data import load_mean_and_std, load_pose_header
-from pose_format.utils.generic import normalize_pose_size
 
 
 def get_pose_appearance(pose: Pose, include_end_frame=False):
@@ -27,7 +26,7 @@ def change_appearace(pose: Pose, appearance: np.ndarray):
     new_pose_data = pose.body.data - appearance
 
     # Bring back the hands
-    hand_components = ['LEFT_HAND_LANDMARKS', 'RIGHT_HAND_LANDMARKS']
+    hand_components = ["LEFT_HAND_LANDMARKS", "RIGHT_HAND_LANDMARKS"]
     for component in pose.header.components:
         if component.name in hand_components:
             # pylint: disable=protected-access
@@ -36,12 +35,12 @@ def change_appearace(pose: Pose, appearance: np.ndarray):
             new_pose_data[:, :, start:end] = pose.body.data[:, :, start:end]
 
     # Bring back the wrists
-    body_component = next(c for c in pose.header.components if c.name == 'POSE_LANDMARKS')
-    points = [f'{hand}_{point}' for hand in ['LEFT', 'RIGHT'] for point in ['WRIST', 'PINKY', 'INDEX', 'THUMB']]
+    body_component = next(c for c in pose.header.components if c.name == "POSE_LANDMARKS")
+    points = [f"{hand}_{point}" for hand in ["LEFT", "RIGHT"] for point in ["WRIST", "PINKY", "INDEX", "THUMB"]]
     existing_points = [p for p in points if p in body_component.points]
     for point in existing_points:
         # pylint: disable=protected-access
-        wrist_index = pose.header._get_point_index('POSE_LANDMARKS', point)
+        wrist_index = pose.header._get_point_index("POSE_LANDMARKS", point)
         new_pose_data[:, :, wrist_index] = pose.body.data[:, :, wrist_index]
 
     pose.body.data = new_pose_data
@@ -69,8 +68,7 @@ def transfer_appearance(pose: Pose, appearance_pose: Pose, include_end_frame=Fal
     pose_components_points = {c.name: c.points for c in pose.header.components}
     appearance_pose = appearance_pose.get_components(pose_components, pose_components_points)
 
-    assert pose.header.total_points() == appearance_pose.header.total_points(), \
-        "Appearance pose missing points"
+    assert pose.header.total_points() == appearance_pose.header.total_points(), "Appearance pose missing points"
 
     pose, appearance = get_pose_appearance(pose, include_end_frame)
     _, new_appearance = get_pose_appearance(appearance_pose, include_end_frame)

pose_anonymization/bin.py

@@ -3,16 +3,16 @@
 import argparse
 
 from pose_format import Pose
-from pose_format.utils.generic import reduce_holistic, correct_wrists
+from pose_format.utils.generic import correct_wrists, reduce_holistic
 
 from pose_anonymization.appearance import remove_appearance, transfer_appearance
 
 
 def get_args():
     parser = argparse.ArgumentParser()
-    parser.add_argument('--input', required=True, type=str, help='path to input pose file')
-    parser.add_argument('--appearance', type=str, help='path to appearance pose file')
-    parser.add_argument('--output', required=True, type=str, help='path to output pose file')
+    parser.add_argument("--input", required=True, type=str, help="path to input pose file")
+    parser.add_argument("--appearance", type=str, help="path to appearance pose file")
+    parser.add_argument("--output", required=True, type=str, help="path to output pose file")
     return parser.parse_args()
 
 
@@ -37,5 +37,5 @@ def main():
         pose.write(f)
 
 
-if __name__ == '__main__':
+if __name__ == "__main__":
     main()

pose_anonymization/data/__init__.py

@@ -2,7 +2,7 @@
 from pathlib import Path
 
 import numpy as np
-from pose_format import PoseHeader, Pose
+from pose_format import Pose, PoseHeader
 from pose_format.numpy import NumPyPoseBody
 from pose_format.utils.reader import BufferReader
 
@@ -18,7 +18,8 @@ def load_pose_header():
 @lru_cache(maxsize=1)
 def load_mean_and_std():
     import json
-    with open(CURRENT_DIR / "pose_normalization.json", "r", encoding="utf-8") as f:
+
+    with open(CURRENT_DIR / "pose_normalization.json", encoding="utf-8") as f:
         data = json.load(f)
 
     mean, std = [], []
@@ -33,6 +34,7 @@ def load_mean_and_std():
 
     return np.array(mean), std
 
+
 @lru_cache(maxsize=1)
 def load_mean_and_std_pose():
     pose_header = load_pose_header()

pose_anonymization/data/calc_mean_std.py

@@ -11,14 +11,14 @@
 
 
 def process_file(file, pose_header: PoseHeader):
-    with open(file, 'rb') as pose_file:
+    with open(file, "rb") as pose_file:
         pose = Pose.read(pose_file.read())
         pose = pre_process_pose(pose, pose_header=pose_header)
     tensor = pose.body.data.filled(0)
 
     frames_sum = np.sum(tensor, axis=(0, 1))
     frames_squared_sum = np.sum(np.square(tensor), axis=(0, 1))
-    unmasked_frames = pose.body.data[:, :, :, 0:1].mask == False
+    unmasked_frames = ~pose.body.data[:, :, :, 0:1].mask
     num_unmasked_frames = np.sum(unmasked_frames, axis=(0, 1))
 
     return frames_sum, frames_squared_sum, num_unmasked_frames
@@ -47,7 +47,7 @@ def main(poses_location: str):
     print(f"Processing {len(files)} files")
 
     # get a single random pose
-    with open(files[0], 'rb') as pose_file:
+    with open(files[0], "rb") as pose_file:
         pose = Pose.read(pose_file.read())
         pose = pre_process_pose(pose)
 
@@ -62,10 +62,7 @@ def main(poses_location: str):
     for component in pose.header.components:
         component_info = {}
         for point in component.points:
-            component_info[point] = {
-                "mean": mean[i].tolist(),
-                "std": std[i].tolist()
-            }
+            component_info[point] = {"mean": mean[i].tolist(), "std": std[i].tolist()}
             i += 1
         mean_std_info[component.name] = component_info
 
@@ -78,9 +75,10 @@ def main(poses_location: str):
 if __name__ == "__main__":
     import argparse
 
-    parser = argparse.ArgumentParser(description='Collect normalization info')
-    parser.add_argument('--dir', type=str, help='Directory containing the pose files',
-                        default="/Volumes/Echo/GCS/sign-mt-poses")
+    parser = argparse.ArgumentParser(description="Collect normalization info")
+    parser.add_argument(
+        "--dir", type=str, help="Directory containing the pose files", default="/Volumes/Echo/GCS/sign-mt-poses"
+    )
 
     args = parser.parse_args()
 

pose_anonymization/data/draw_mean_pose.py

@@ -13,10 +13,7 @@
     unshift_hands(mean_pose)
     normalize_pose_size(mean_pose)
 
-    poses = {
-        "full": mean_pose,
-        "reduced": reduce_holistic(mean_pose)
-    }
+    poses = {"full": mean_pose, "reduced": reduce_holistic(mean_pose)}
     for name, pose in poses.items():
         pose.focus()
 

pose_anonymization/data/normalization.py

@@ -1,15 +1,15 @@
 from pose_format import Pose, PoseHeader
-from pose_format.utils.generic import pose_normalization_info, correct_wrists, hands_components
+from pose_format.utils.generic import correct_wrists, hands_components, pose_normalization_info
 
-from pose_anonymization.data import load_mean_and_std, load_mean_and_std_pose
+from pose_anonymization.data import load_mean_and_std_pose
 
 
 def shift_hand(pose: Pose, hand_component: str, wrist_name: str):
     # pylint: disable=protected-access
     wrist_index = pose.header._get_point_index(hand_component, wrist_name)
-    hand = pose.body.data[:, :, wrist_index: wrist_index + 21]
+    hand = pose.body.data[:, :, wrist_index : wrist_index + 21]
     wrist = hand[:, :, 0:1]
-    pose.body.data[:, :, wrist_index: wrist_index + 21] = hand - wrist
+    pose.body.data[:, :, wrist_index : wrist_index + 21] = hand - wrist
 
 
 def shift_hands(pose: Pose):
@@ -21,12 +21,12 @@ def shift_hands(pose: Pose):
 def unshift_hand(pose: Pose, hand_component: str):
     # pylint: disable=protected-access
     wrist_index = pose.header._get_point_index(hand_component, "WRIST")
-    hand = pose.body.data[:, :, wrist_index: wrist_index + 21]
+    hand = pose.body.data[:, :, wrist_index : wrist_index + 21]
     body_wrist_name = "LEFT_WRIST" if hand_component == "LEFT_HAND_LANDMARKS" else "RIGHT_WRIST"
     # pylint: disable=protected-access
     body_wrist_index = pose.header._get_point_index("POSE_LANDMARKS", body_wrist_name)
-    body_wrist = pose.body.data[:, :, body_wrist_index: body_wrist_index + 1]
-    pose.body.data[:, :, wrist_index: wrist_index + 21] = hand + body_wrist
+    body_wrist = pose.body.data[:, :, body_wrist_index : body_wrist_index + 1]
+    pose.body.data[:, :, wrist_index : wrist_index + 21] = hand + body_wrist
 
 
 def unshift_hands(pose: Pose):
@@ -58,8 +58,7 @@ def load_mean_and_std_for_pose(pose: Pose):
     mean_pose, std_pose = load_mean_and_std_pose()
     mean_pose = pose_like(mean_pose, pose.header)
     std_pose = pose_like(std_pose, pose.header)
-    return (mean_pose.body.data.reshape((-1, 3)),
-            std_pose.body.data.reshape((-1, 3)))
+    return (mean_pose.body.data.reshape((-1, 3)), std_pose.body.data.reshape((-1, 3)))
 
 
 def normalize_mean_std(pose: Pose):

pose_anonymization/evaluate_example_stitching.py

@@ -16,7 +16,7 @@
 for directory in [example_dir / "original", example_dir / "anonymized"]:
     poses = []
     for word in ["kleine", "kinder", "essen", "pizza"]:
-        with open(directory / f"{word}.pose", 'rb') as pose_file:
+        with open(directory / f"{word}.pose", "rb") as pose_file:
             poses.append(Pose.read(pose_file.read()))
 
     concatenated = concatenate_poses(poses).get_components(["FACE_LANDMARKS"])
@@ -27,7 +27,7 @@
     plt.plot(flow, label=directory.name.capitalize())
 
 plt.yticks([])
-plt.legend(loc='upper left')
+plt.legend(loc="upper left")
 plt.ylabel("Optical Flow")
 plt.tight_layout()
 fig.show()