bug fix

Author: voorhs

autointent/_dump_tools/base.py

@@ -31,14 +31,14 @@
 class BaseObjectDumper(ABC, Generic[T]):
     dir_or_file_name: str
 
-    @abstractmethod
     @staticmethod
+    @abstractmethod
     def dump(obj: T, path: Path, exists_ok: bool) -> None: ...
 
-    @abstractmethod
     @staticmethod
+    @abstractmethod
     def load(path: Path, **kwargs: Any) -> T: ...  # noqa: ANN401
 
-    @abstractmethod
     @classmethod
+    @abstractmethod
     def check_isinstance(cls, obj: Any) -> bool: ...  # noqa: ANN401

autointent/_dump_tools/main.py

@@ -7,6 +7,7 @@
 
 from autointent.configs import CrossEncoderConfig, EmbedderConfig
 from autointent.context.optimization_info import Artifact
+from autointent.schemas import TagsList
 
 from .base import BaseObjectDumper, ModuleAttributes, ModuleSimpleAttributes
 from .unit_dumpers import (
@@ -101,7 +102,7 @@ def dump(
                 continue
 
             # Handle simple attributes and arrays separately
-            if isinstance(val, ModuleSimpleAttributes):
+            if isinstance(val, ModuleSimpleAttributes) and not isinstance(val, TagsList):
                 simple_attrs[key] = val
             elif isinstance(val, np.ndarray):
                 arrays[key] = val

autointent/_dump_tools/unit_dumpers.py

@@ -120,7 +120,8 @@ class EstimatorDumper(BaseObjectDumper[BaseEstimator]):
     dir_or_file_name = "estimators"
 
     @staticmethod
-    def dump(obj: BaseEstimator, path: Path, exists_ok: bool) -> None:  # noqa: ARG004
+    def dump(obj: BaseEstimator, path: Path, exists_ok: bool) -> None:
+        path.parent.mkdir(parents=True, exist_ok=exists_ok)
         joblib.dump(obj, path)
 
     @staticmethod

autointent/schemas/_schemas.py

@@ -31,6 +31,7 @@ def __init__(self, tags: list[Tag]) -> None:
         super().__init__(tags)
 
     def dump(self, path: Path) -> None:
+        path.parent.mkdir(parents=True, exist_ok=True)
         serialized = [v.model_dump(mode="json") for v in self]
         with path.open("w", encoding="utf-8") as file:
             json.dump(serialized, file, indent=4, ensure_ascii=False)