Merge branch 'dev' into docs

Author: voorhs

autointent/modules/__init__.py

@@ -47,6 +47,7 @@ def create_modules_dict(modules: list[type[T]]) -> dict[str, type[T]]:
     "RETRIEVAL_MODULES_MULTILABEL",
     "SCORING_MODULES_MULTICLASS",
     "SCORING_MODULES_MULTILABEL",
+    "Module",
     "AdaptivePredictor",
     "ArgmaxPredictor",
     "DNNCScorer",

autointent/modules/prediction/adaptive.py

@@ -28,7 +28,6 @@ class AdaptivePredictorDumpMetadata(TypedDict):
     :ivar tags: List of Tag objects for mutually exclusive classes.
     :ivar n_classes: Number of classes used during training.
     """
-
     r: float
     tags: list[Tag] | None
     n_classes: int
@@ -47,7 +46,6 @@ class AdaptivePredictor(PredictionModule):
     :ivar tags: List of Tag objects for mutually exclusive classes.
     :ivar name: Name of the predictor, defaults to "adaptive".
     """
-
     metadata_dict_name = "metadata.json"
     n_classes: int
     _r: float

autointent/modules/prediction/argmax.py

@@ -18,13 +18,11 @@
 
 class ArgmaxPredictorDumpMetadata(BaseMetadataDict):
     """Argmax predictor metadata."""
-
     n_classes: int
 
 
 class ArgmaxPredictor(PredictionModule):
     """Argmax prediction module."""
-
     name = "argmax"
     n_classes: int
 

autointent/modules/prediction/jinoos.py

@@ -70,7 +70,6 @@ def fit(
         :param tags: Tags to fit
         """
         # TODO: use dev split instead of test split.
-
         multilabel = isinstance(labels[0], list)
         if multilabel:
             msg = "JinoosPredictor is compatible with single-label classification only"

autointent/modules/prediction/utils.py

@@ -1,5 +1,4 @@
 """Utility functions and custom exceptions for handling multilabel predictions and errors."""
-
 from typing import Any
 
 import numpy as np