Skip to content

Commit

Permalink
lint
Browse files Browse the repository at this point in the history
  • Loading branch information
Borda committed Jul 22, 2024
1 parent 10c6609 commit 9ae3ee3
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 5 deletions.
2 changes: 1 addition & 1 deletion src/torchmetrics/classification/logauc.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,7 @@ def __init__(
fpr_range: Tuple[float, float] = (0.001, 0.1),
thresholds: Optional[Union[float, Tensor]] = None,
ignore_index: Optional[int] = None,
validate_args: bool = None,
validate_args: bool = False,
**kwargs: Any,
) -> None:
super().__init__(thresholds=thresholds, ignore_index=ignore_index, validate_args=validate_args, **kwargs)
Expand Down
8 changes: 4 additions & 4 deletions tests/unittests/classification/test_logauc.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,14 +19,14 @@
from scipy.special import expit as sigmoid
from scipy.special import softmax
from tdc.evaluator import range_logAUC
from torchmetrics.classification.logauc import BinaryLogAUC, MulticlassLogAUC, MultilabelLogAUC
from torchmetrics.functional.classification.logauc import binary_logauc, multiclass_logauc, multilabel_logauc
from torchmetrics.classification.logauc import BinaryLogAUC, MulticlassLogAUC
from torchmetrics.functional.classification.logauc import binary_logauc, multiclass_logauc
from torchmetrics.functional.classification.roc import binary_roc

from unittests import NUM_CLASSES
from unittests.classification.inputs import _binary_cases, _multiclass_cases, _multilabel_cases
from unittests.classification.inputs import _binary_cases, _multiclass_cases
from unittests.helpers import seed_all
from unittests.helpers.testers import MetricTester, inject_ignore_index, remove_ignore_index
from unittests.helpers.testers import MetricTester

seed_all(42)

Expand Down

0 comments on commit 9ae3ee3

Please sign in to comment.