Skip to content

Commit 545fbe4

Browse files
committed
fix codestyle
1 parent 5c217a9 commit 545fbe4

File tree

4 files changed

+5
-1
lines changed

4 files changed

+5
-1
lines changed

autointent/modules/__init__.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,6 @@ def create_modules_dict(modules: list[type[T]]) -> dict[str, type[T]]:
4747
"RETRIEVAL_MODULES_MULTILABEL",
4848
"SCORING_MODULES_MULTICLASS",
4949
"SCORING_MODULES_MULTILABEL",
50-
"Module",
5150
"AdaptivePredictor",
5251
"ArgmaxPredictor",
5352
"DNNCScorer",

autointent/modules/prediction/adaptive.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@ class AdaptivePredictorDumpMetadata(TypedDict):
2828
:ivar tags: List of Tag objects for mutually exclusive classes.
2929
:ivar n_classes: Number of classes used during training.
3030
"""
31+
3132
r: float
3233
tags: list[Tag] | None
3334
n_classes: int
@@ -46,6 +47,7 @@ class AdaptivePredictor(PredictionModule):
4647
:ivar tags: List of Tag objects for mutually exclusive classes.
4748
:ivar name: Name of the predictor, defaults to "adaptive".
4849
"""
50+
4951
metadata_dict_name = "metadata.json"
5052
n_classes: int
5153
_r: float

autointent/modules/prediction/argmax.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,11 +18,13 @@
1818

1919
class ArgmaxPredictorDumpMetadata(BaseMetadataDict):
2020
"""Argmax predictor metadata."""
21+
2122
n_classes: int
2223

2324

2425
class ArgmaxPredictor(PredictionModule):
2526
"""Argmax prediction module."""
27+
2628
name = "argmax"
2729
n_classes: int
2830

autointent/modules/prediction/utils.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
"""Utility functions and custom exceptions for handling multilabel predictions and errors."""
2+
23
from typing import Any
34

45
import numpy as np

0 commit comments

Comments
 (0)