updated installation and added github actions

.github/workflows/formatting.yaml

@@ -0,0 +1,23 @@
+name: Formatting
+on: [push, pull_request]
+jobs:
+  check-formatting:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v1
+      - name: Black Code Formatter
+        uses: lgeiger/black-action@master
+        with:
+          args: "--line-length 120 . --check"
+          name: black-action
+
+  auto-formatting:
+    name: runner / black
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v2
+      - name: Check files using the black formatter
+        uses: rickstaa/action-black@v1
+        id: action_black
+        with:
+          black_args: "--line-length 120 ."

.github/workflows/pytest.yaml

@@ -0,0 +1,15 @@
+steps:
+- uses: actions/checkout@v4
+- name: Set up Python
+  uses: actions/setup-python@v4
+  with:
+    python-version: '3.x'
+- name: Install dependencies
+  run: |
+    python -m pip install --upgrade pip
+    pip install -e ./
+    pip3 install torch torchvision torchaudio
+    pip install pytest
+- name: Test with pytest
+  run: |
+    pytest ./

README.md

@@ -32,6 +32,7 @@ Code for ICML 2024: <a href="some_ariv_link" target="_blank">"Position Paper: Le
 ```shell
 git clone git@github.com:martius-lab/hitchhiking-rotations.git
 pip3 install -e ./
+pip3 install torch torchvision torchaudio
 ```
 
 # Experiments

setup.py

@@ -1,7 +1,7 @@
 from setuptools import find_packages
 from distutils.core import setup
 
-INSTALL_REQUIRES = ["numpy", "pip", "scipy", "scikit-learn", "matplotlib", "seaborn", "pytictac", "roma", "black"]
+INSTALL_REQUIRES = ["numpy", "pip", "scipy", "scikit-learn", "matplotlib", "seaborn", "pytictac", "roma", "black", "pyyaml", "hydra", "omegaconf", "tqdm", "hydra-core"]
 
 setup(
     name="hitchhiking_rotations",

tests/roma_verify_rotations.py → tests/test_roma_verify_rotations.py

@@ -2,7 +2,7 @@
 import roma
 import numpy as np
 from scipy.spatial.transform import Rotation as R
-from pose_estimation import euler_angles_to_matrix
+from hitchhiking_rotations.utils import euler_angles_to_matrix
 
 
 def test_roma_quaternion():