From 17a2ceac2359955e496ece0c9ac59fd460622e22 Mon Sep 17 00:00:00 2001 From: Toraudonn Date: Tue, 22 Jun 2021 10:24:58 +0000 Subject: [PATCH] linting, formatting, readme changes --- README.md | 2 ++ equilib/__init__.py | 2 +- equilib/equi2equi/base.py | 2 +- equilib/equi2pers/equi2pers_numpy.py | 6 ++++-- equilib/equi2pers/equi2pers_torch.py | 4 +++- 5 files changed, 11 insertions(+), 5 deletions(-) diff --git a/README.md b/README.md index 92f14c41..3a6436b2 100644 --- a/README.md +++ b/README.md @@ -40,6 +40,8 @@ cd equilib pip install -r requirements.txt +pip install -e . +# or python setup.py develop ``` diff --git a/equilib/__init__.py b/equilib/__init__.py index 4c8c0f75..4ea94725 100644 --- a/equilib/__init__.py +++ b/equilib/__init__.py @@ -4,7 +4,7 @@ from equilib.equi2cube.base import Equi2Cube, equi2cube from equilib.equi2equi.base import Equi2Equi, equi2equi from equilib.equi2pers.base import Equi2Pers, equi2pers -from equilib.info import __version__ +from equilib.info import __version__ # noqa __all__ = [ "Cube2Equi", diff --git a/equilib/equi2equi/base.py b/equilib/equi2equi/base.py index 38744e9c..e6b409c6 100644 --- a/equilib/equi2equi/base.py +++ b/equilib/equi2equi/base.py @@ -34,7 +34,7 @@ def __init__( h_out: Optional[int] = None, sampling_method: str = "default", mode: str = "bilinear", - z_down: bool = False + z_down: bool = False, ) -> None: self.w_out = w_out self.h_out = h_out diff --git a/equilib/equi2pers/equi2pers_numpy.py b/equilib/equi2pers/equi2pers_numpy.py index 7293ed58..2c0ada18 100644 --- a/equilib/equi2pers/equi2pers_numpy.py +++ b/equilib/equi2pers/equi2pers_numpy.py @@ -9,10 +9,12 @@ from equilib.utils import ( create_global2camera_rotation_matrix, create_intrinsic_matrix, - create_rotation_matrix + create_rotation_matrix, ) -_create_global2camera_rotation_matrix = partial(create_global2camera_rotation_matrix, is_torch=False) +_create_global2camera_rotation_matrix = partial( + create_global2camera_rotation_matrix, is_torch=False +) _create_intrinsic_matrix = partial(create_intrinsic_matrix, is_torch=False) _create_rotation_matrix = partial(create_rotation_matrix, is_torch=False) diff --git a/equilib/equi2pers/equi2pers_torch.py b/equilib/equi2pers/equi2pers_torch.py index 3d2ce361..0ef22a52 100644 --- a/equilib/equi2pers/equi2pers_torch.py +++ b/equilib/equi2pers/equi2pers_torch.py @@ -14,7 +14,9 @@ torch_utils, ) -_create_global2camera_rotation_matrix = partial(create_global2camera_rotation_matrix, is_torch=True) +_create_global2camera_rotation_matrix = partial( + create_global2camera_rotation_matrix, is_torch=True +) _create_intrinsic_matrix = partial(create_intrinsic_matrix, is_torch=True) _create_rotation_matrix = partial(create_rotation_matrix, is_torch=True)