simplify fields

Author: Samoed

autointent/_pipeline/_pipeline.py

@@ -13,10 +13,10 @@
 from autointent.custom_types import ListOfGenericLabels, NodeType
 from autointent.metrics import PREDICTION_METRICS_MULTILABEL
 from autointent.nodes import InferenceNode, NodeOptimizer
+from autointent.nodes.schemes import OptimizerConfig
 from autointent.utils import load_default_search_space, load_search_space
 
 from ._schemas import InferencePipelineOutput, InferencePipelineUtteranceOutput
-from ..nodes.schemes import OptimizerConfig
 
 if TYPE_CHECKING:
     from autointent.modules.abc import DecisionModule, ScoringModule
@@ -77,8 +77,8 @@ def from_search_space(cls, search_space: list[dict[str, Any]] | Path | str, seed
         """
         if isinstance(search_space, Path | str):
             search_space = load_search_space(search_space)
-        search_space = OptimizerConfig(search_space).model_dump()
-        nodes = [NodeOptimizer(**node) for node in search_space]
+        validated_search_space = OptimizerConfig(search_space).model_dump()  # type: ignore[arg-type]
+        nodes = [NodeOptimizer(**node) for node in validated_search_space]
         return cls(nodes=nodes, seed=seed)
 
     @classmethod

autointent/modules/decision/_threshold.py

@@ -82,7 +82,7 @@ def __init__(
 
         :param thresh: Threshold for the scores, shape (n_classes,) or float
         """
-        self.thresh = thresh
+        self.thresh = thresh if isinstance(thresh, float) else np.array(thresh)
 
     @classmethod
     def from_context(cls, context: Context, thresh: float | list[float] = 0.5) -> "ThresholdDecision":

autointent/nodes/schemes.py

@@ -29,13 +29,9 @@ def generate_models_and_union_type_for_classes(
                 continue
 
             param_type: TypeAlias = type_hints.get(param_name, Any)  # type: ignore[valid-type]  # noqa: PYI042
-            default = param.default if param.default is not inspect.Parameter.empty else None
+            field = Field(default=[param.default]) if param.default is not inspect.Parameter.empty else Field(...)
 
-            # Ensure fields with defaults have the correct type
-            if default is None:
-                param_type = param_type | None
-
-            fields[param_name] = (list[param_type], Field(default=default))  # type: ignore[assignment]
+            fields[param_name] = (list[param_type], field)  # type: ignore[assignment]
 
         model_name = f"{cls.__name__}InitModel"
         models[cls.__name__] = type(

tests/configs/test_combined_config.py

@@ -51,7 +51,7 @@ def test_valid_optimizer_config(valid_optimizer_config):
     "task_type",
     ["multiclass", "multilabel"],
 )
-def test_inference_config(task_type):
+def test_optimizer_config(task_type):
     search_space = get_search_space(task_type)
     config = OptimizerConfig(search_space)
     assert config