Skip to content

Commit

Permalink
linting, formatting, readme changes
Browse files Browse the repository at this point in the history
  • Loading branch information
haruishi43 committed Jun 22, 2021
1 parent e80fdd0 commit 17a2cea
Show file tree
Hide file tree
Showing 5 changed files with 11 additions and 5 deletions.
2 changes: 2 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -40,6 +40,8 @@ cd equilib

pip install -r requirements.txt

pip install -e .
# or
python setup.py develop
```

Expand Down
2 changes: 1 addition & 1 deletion equilib/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down
2 changes: 1 addition & 1 deletion equilib/equi2equi/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
6 changes: 4 additions & 2 deletions equilib/equi2pers/equi2pers_numpy.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)

Expand Down
4 changes: 3 additions & 1 deletion equilib/equi2pers/equi2pers_torch.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)

Expand Down

0 comments on commit 17a2cea

Please sign in to comment.