Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Jan 14, 2022
1 parent 2b0d107 commit 9866e0d
Show file tree
Hide file tree
Showing 3 changed files with 13 additions and 19 deletions.
3 changes: 1 addition & 2 deletions torchmetrics/image/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,8 +17,7 @@
MultiScaleStructuralSimilarityIndexMeasure,
StructuralSimilarityIndexMeasure,
)

from torchmetrics.utilities.imports import _TORCH_FIDELITY_AVAILABLE

if _TORCH_FIDELITY_AVAILABLE:
from torchmetrics.image.fid import FrechetInceptionDistance
from torchmetrics.image.fid import FrechetInceptionDistance
8 changes: 1 addition & 7 deletions torchmetrics/image/fid.py
Original file line number Diff line number Diff line change
Expand Up @@ -316,10 +316,4 @@ def __init__(
process_group: Optional[Any] = None,
dist_sync_fn: Callable[[Tensor], List[Tensor]] = None,
) -> None:
void(
feature,
compute_on_step,
dist_sync_on_step,
process_group,
dist_sync_fn
)
void(feature, compute_on_step, dist_sync_on_step, process_group, dist_sync_fn)
21 changes: 11 additions & 10 deletions torchmetrics/image/kid.py
Original file line number Diff line number Diff line change
Expand Up @@ -314,14 +314,15 @@ def __init__(
process_group: Optional[Any] = None,
dist_sync_fn: Callable = None,
) -> None:
void(feature,
subsets,
subset_size,
degree,
gamma,
coef,
compute_on_step,
dist_sync_on_step,
process_group,
dist_sync_fn
void(
feature,
subsets,
subset_size,
degree,
gamma,
coef,
compute_on_step,
dist_sync_on_step,
process_group,
dist_sync_fn,
)

0 comments on commit 9866e0d

Please sign in to comment.