run black

martius-lab · Mar 1, 2024 · aeaf838 · aeaf838
1 parent b6d50de
commit aeaf838
Show file tree

Hide file tree

Showing 24 changed files with 62 additions and 47 deletions.
diff --git a/hitchhiking_rotations/__init__.py b/hitchhiking_rotations/__init__.py
@@ -1,8 +1,8 @@
-#                                                                               
+#
 # Copyright (c) 2024, MPI-IS, Jonas Frey, Rene Geist, Mikel Zhobro.
 # All rights reserved. Licensed under the MIT license.
 # See LICENSE file in the project root for details.
-#                                                                               
+#
 import os
 
 HITCHHIKING_ROOT_DIR = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))

diff --git a/hitchhiking_rotations/cfgs/__init__.py b/hitchhiking_rotations/cfgs/__init__.py
@@ -1,8 +1,8 @@
-#                                                                               
+#
 # Copyright (c) 2024, MPI-IS, Jonas Frey, Rene Geist, Mikel Zhobro.
 # All rights reserved. Licensed under the MIT license.
 # See LICENSE file in the project root for details.
-#                                                                               
+#
 from .cfg_cube_image_to_pose import get_cfg_cube_image_to_pose
 from .cfg_pcd_to_pose import get_cfg_pcd_to_pose
 from .cfg_pose_to_cube_image import get_cfg_pose_to_cube_image
diff --git a/hitchhiking_rotations/cfgs/cfg_cube_image_to_pose.py b/hitchhiking_rotations/cfgs/cfg_cube_image_to_pose.py
@@ -1,8 +1,8 @@
-#                                                                               
+#
 # Copyright (c) 2024, MPI-IS, Jonas Frey, Rene Geist, Mikel Zhobro.
 # All rights reserved. Licensed under the MIT license.
 # See LICENSE file in the project root for details.
-#                                                                               
+#
 def get_cfg_cube_image_to_pose(device):
     shared_trainer_cfg = {
         "_target_": "hitchhiking_rotations.utils.Trainer",

diff --git a/hitchhiking_rotations/cfgs/cfg_pcd_to_pose.py b/hitchhiking_rotations/cfgs/cfg_pcd_to_pose.py
@@ -1,8 +1,8 @@
-#                                                                               
+#
 # Copyright (c) 2024, MPI-IS, Jonas Frey, Rene Geist, Mikel Zhobro.
 # All rights reserved. Licensed under the MIT license.
 # See LICENSE file in the project root for details.
-#                                                                               
+#
 def get_cfg_pcd_to_pose(device):
     shared_trainer_cfg = {
         "_target_": "hitchhiking_rotations.utils.Trainer",

diff --git a/hitchhiking_rotations/cfgs/cfg_pose_to_cube_image.py b/hitchhiking_rotations/cfgs/cfg_pose_to_cube_image.py
@@ -1,8 +1,8 @@
-#                                                                               
+#
 # Copyright (c) 2024, MPI-IS, Jonas Frey, Rene Geist, Mikel Zhobro.
 # All rights reserved. Licensed under the MIT license.
 # See LICENSE file in the project root for details.
-#                                                                               
+#
 def get_cfg_pose_to_cube_image(device):
     cfg = {
         "_target_": "hitchhiking_rotations.utils.Trainer",

diff --git a/hitchhiking_rotations/cfgs/cfg_pose_to_fourier.py b/hitchhiking_rotations/cfgs/cfg_pose_to_fourier.py
@@ -1,8 +1,8 @@
-#                                                                               
+#
 # Copyright (c) 2024, MPI-IS, Jonas Frey, Rene Geist, Mikel Zhobro.
 # All rights reserved. Licensed under the MIT license.
 # See LICENSE file in the project root for details.
-#                                                                               
+#
 def get_cfg_pose_to_fourier(device, nb, nf):
     shared_trainer_cfg = {
         "_target_": "hitchhiking_rotations.utils.Trainer",

diff --git a/hitchhiking_rotations/datasets/__init__.py b/hitchhiking_rotations/datasets/__init__.py
@@ -1,8 +1,8 @@
-#                                                                               
+#
 # Copyright (c) 2024, MPI-IS, Jonas Frey, Rene Geist, Mikel Zhobro.
 # All rights reserved. Licensed under the MIT license.
 # See LICENSE file in the project root for details.
-#                                                                               
+#
 from .cube_dataset import CubeImageToPoseDataset, PoseToCubeImageDataset
 from .pointcloud_dataset import PointCloudDataset
 from .fourier_dataset import PoseToFourierDataset
diff --git a/hitchhiking_rotations/datasets/cube_data_generator.py b/hitchhiking_rotations/datasets/cube_data_generator.py
@@ -1,8 +1,8 @@
-#                                                                               
+#
 # Copyright (c) 2024, MPI-IS, Jonas Frey, Rene Geist, Mikel Zhobro.
 # All rights reserved. Licensed under the MIT license.
 # See LICENSE file in the project root for details.
-#                                                                               
+#
 import mujoco
 import torch
 import numpy as np

diff --git a/hitchhiking_rotations/datasets/cube_dataset.py b/hitchhiking_rotations/datasets/cube_dataset.py
@@ -1,8 +1,8 @@
-#                                                                               
+#
 # Copyright (c) 2024, MPI-IS, Jonas Frey, Rene Geist, Mikel Zhobro.
 # All rights reserved. Licensed under the MIT license.
 # See LICENSE file in the project root for details.
-#                                                                               
+#
 from hitchhiking_rotations import HITCHHIKING_ROOT_DIR
 from hitchhiking_rotations.utils import save_pickle, load_pickle
 import os

diff --git a/hitchhiking_rotations/datasets/fourier_dataset.py b/hitchhiking_rotations/datasets/fourier_dataset.py
@@ -1,8 +1,8 @@
-#                                                                               
+#
 # Copyright (c) 2024, MPI-IS, Jonas Frey, Rene Geist, Mikel Zhobro.
 # All rights reserved. Licensed under the MIT license.
 # See LICENSE file in the project root for details.
-#                                                                               
+#
 from torch.utils.data import Dataset
 
 

diff --git a/hitchhiking_rotations/datasets/pointcloud_dataset.py b/hitchhiking_rotations/datasets/pointcloud_dataset.py
@@ -1,8 +1,8 @@
-#                                                                               
+#
 # Copyright (c) 2024, MPI-IS, Jonas Frey, Rene Geist, Mikel Zhobro.
 # All rights reserved. Licensed under the MIT license.
 # See LICENSE file in the project root for details.
-#                                                                               
+#
 import torch
 import numpy as np
 import os

diff --git a/hitchhiking_rotations/models/__init__.py b/hitchhiking_rotations/models/__init__.py
@@ -1,6 +1,6 @@
-#                                                                               
+#
 # Copyright (c) 2024, MPI-IS, Jonas Frey, Rene Geist, Mikel Zhobro.
 # All rights reserved. Licensed under the MIT license.
 # See LICENSE file in the project root for details.
-#                                                                               
+#
 from .models import MLP, CNN, MLPNetPCD
diff --git a/hitchhiking_rotations/models/models.py b/hitchhiking_rotations/models/models.py
@@ -1,8 +1,8 @@
-#                                                                               
+#
 # Copyright (c) 2024, MPI-IS, Jonas Frey, Rene Geist, Mikel Zhobro.
 # All rights reserved. Licensed under the MIT license.
 # See LICENSE file in the project root for details.
-#                                                                               
+#
 import numpy as np
 import torch
 from torch import nn

diff --git a/hitchhiking_rotations/utils/__init__.py b/hitchhiking_rotations/utils/__init__.py
@@ -1,8 +1,8 @@
-#                                                                               
+#
 # Copyright (c) 2024, MPI-IS, Jonas Frey, Rene Geist, Mikel Zhobro.
 # All rights reserved. Licensed under the MIT license.
 # See LICENSE file in the project root for details.
-#                                                                               
+#
 from .euler_helper import euler_angles_to_matrix, matrix_to_euler_angles
 from .conversions import *
 from .metrics import *

diff --git a/hitchhiking_rotations/utils/conversions.py b/hitchhiking_rotations/utils/conversions.py
@@ -1,8 +1,8 @@
-#                                                                               
+#
 # Copyright (c) 2024, MPI-IS, Jonas Frey, Rene Geist, Mikel Zhobro.
 # All rights reserved. Licensed under the MIT license.
 # See LICENSE file in the project root for details.
-#                                                                               
+#
 from .euler_helper import euler_angles_to_matrix, matrix_to_euler_angles
 import roma
 import torch

diff --git a/hitchhiking_rotations/utils/euler_helper.py b/hitchhiking_rotations/utils/euler_helper.py
@@ -1,8 +1,8 @@
-#                                                                               
+#
 # Copyright (c) 2024, MPI-IS, Jonas Frey, Rene Geist, Mikel Zhobro.
 # All rights reserved. Licensed under the MIT license.
 # See LICENSE file in the project root for details.
-#                                                                               
+#
 import torch
 
 

diff --git a/hitchhiking_rotations/utils/helper.py b/hitchhiking_rotations/utils/helper.py
@@ -1,8 +1,8 @@
-#                                                                               
+#
 # Copyright (c) 2024, MPI-IS, Jonas Frey, Rene Geist, Mikel Zhobro.
 # All rights reserved. Licensed under the MIT license.
 # See LICENSE file in the project root for details.
-#                                                                               
+#
 def passthrough(*x):
     if len(x) == 1:
         return x[0]

diff --git a/hitchhiking_rotations/utils/loading.py b/hitchhiking_rotations/utils/loading.py
@@ -1,8 +1,8 @@
-#                                                                               
+#
 # Copyright (c) 2024, MPI-IS, Jonas Frey, Rene Geist, Mikel Zhobro.
 # All rights reserved. Licensed under the MIT license.
 # See LICENSE file in the project root for details.
-#                                                                               
+#
 import os
 import yaml
 import pickle

diff --git a/hitchhiking_rotations/utils/logger.py b/hitchhiking_rotations/utils/logger.py
@@ -1,8 +1,8 @@
-#                                                                               
+#
 # Copyright (c) 2024, MPI-IS, Jonas Frey, Rene Geist, Mikel Zhobro.
 # All rights reserved. Licensed under the MIT license.
 # See LICENSE file in the project root for details.
-#                                                                               
+#
 from .metrics import *
 
 available_metrics = {

diff --git a/hitchhiking_rotations/utils/metrics.py b/hitchhiking_rotations/utils/metrics.py
@@ -1,8 +1,8 @@
-#                                                                               
+#
 # Copyright (c) 2024, MPI-IS, Jonas Frey, Rene Geist, Mikel Zhobro.
 # All rights reserved. Licensed under the MIT license.
 # See LICENSE file in the project root for details.
-#                                                                               
+#
 import torch
 import roma
 

diff --git a/hitchhiking_rotations/utils/notation.py b/hitchhiking_rotations/utils/notation.py
@@ -1,8 +1,8 @@
-#                                                                               
+#
 # Copyright (c) 2024, MPI-IS, Jonas Frey, Rene Geist, Mikel Zhobro.
 # All rights reserved. Licensed under the MIT license.
 # See LICENSE file in the project root for details.
-#                                                                               
+#
 from enum import Enum
 
 

diff --git a/hitchhiking_rotations/utils/trainer.py b/hitchhiking_rotations/utils/trainer.py
@@ -1,8 +1,8 @@
-#                                                                               
+#
 # Copyright (c) 2024, MPI-IS, Jonas Frey, Rene Geist, Mikel Zhobro.
 # All rights reserved. Licensed under the MIT license.
 # See LICENSE file in the project root for details.
-#                                                                               
+#
 import torch
 
 

diff --git a/hitchhiking_rotations/utils/training_helper.py b/hitchhiking_rotations/utils/training_helper.py
@@ -1,8 +1,8 @@
-#                                                                               
+#
 # Copyright (c) 2024, MPI-IS, Jonas Frey, Rene Geist, Mikel Zhobro.
 # All rights reserved. Licensed under the MIT license.
 # See LICENSE file in the project root for details.
-#                                                                               
+#
 from enum import Enum
 import numpy as np
 import argparse

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