Skip to content

Commit

Permalink
remove check_compute_fn argument
Browse files Browse the repository at this point in the history
  • Loading branch information
kzkadc committed Sep 10, 2024
1 parent c0a05a0 commit 642b3f6
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 4 deletions.
3 changes: 1 addition & 2 deletions ignite/metrics/regression/kendall_correlation.py
Original file line number Diff line number Diff line change
Expand Up @@ -90,7 +90,6 @@ def __init__(
self,
variant: str = "b",
output_transform: Callable[..., Any] = lambda x: x,
check_compute_fn: bool = True,
device: Union[str, torch.device] = torch.device("cpu"),
skip_unrolling: bool = False,
) -> None:
Expand All @@ -99,7 +98,7 @@ def __init__(
except ImportError:
raise ModuleNotFoundError("This module requires scipy to be installed.")

super().__init__(_get_kendall_tau(variant), output_transform, check_compute_fn, device, skip_unrolling)
super().__init__(_get_kendall_tau(variant), output_transform, True, device, skip_unrolling)

def update(self, output: Tuple[torch.Tensor, torch.Tensor]) -> None:
y_pred, y = output[0].detach(), output[1].detach()
Expand Down
3 changes: 1 addition & 2 deletions ignite/metrics/regression/spearman_correlation.py
Original file line number Diff line number Diff line change
Expand Up @@ -80,7 +80,6 @@ class SpearmanRankCorrelation(EpochMetric):
def __init__(
self,
output_transform: Callable[..., Any] = lambda x: x,
check_compute_fn: bool = True,
device: Union[str, torch.device] = torch.device("cpu"),
skip_unrolling: bool = False,
) -> None:
Expand All @@ -89,7 +88,7 @@ def __init__(
except ImportError:
raise ModuleNotFoundError("This module requires scipy to be installed.")

super().__init__(_get_spearman_r(), output_transform, check_compute_fn, device, skip_unrolling)
super().__init__(_get_spearman_r(), output_transform, True, device, skip_unrolling)

def update(self, output: Tuple[torch.Tensor, torch.Tensor]) -> None:
y_pred, y = output[0].detach(), output[1].detach()
Expand Down

0 comments on commit 642b3f6

Please # to comment.