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 Nov 30, 2021
1 parent be20ddb commit e57142f
Showing 1 changed file with 10 additions and 2 deletions.
12 changes: 10 additions & 2 deletions torchmetrics/metric.py
Original file line number Diff line number Diff line change
Expand Up @@ -759,9 +759,17 @@ def compute(self) -> Any:
@torch.jit.unused
def forward(self, *args: Any, **kwargs: Any) -> Any:

val_a = self.metric_a(*args, **self.metric_a._filter_kwargs(**kwargs)) if isinstance(self.metric_a, Metric) else self.metric_a
val_a = (
self.metric_a(*args, **self.metric_a._filter_kwargs(**kwargs))
if isinstance(self.metric_a, Metric)
else self.metric_a
)

val_b = self.metric_b(*args, **self.metric_b._filter_kwargs(**kwargs)) if isinstance(self.metric_b, Metric) else self.metric_b
val_b = (
self.metric_b(*args, **self.metric_b._filter_kwargs(**kwargs))
if isinstance(self.metric_b, Metric)
else self.metric_b
)

if val_a is None:
# compute_on_step of metric_a is False
Expand Down

0 comments on commit e57142f

Please sign in to comment.