sinapsis-anomalib 0.1.9__tar.gz → 0.1.11__tar.gz

{sinapsis_anomalib-0.1.9/src/sinapsis_anomalib.egg-info → sinapsis_anomalib-0.1.11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sinapsis-anomalib
-Version: 0.1.9
+Version: 0.1.11
 Summary: Templates for anomaly detection with computer vision using anomalib library.
 Author-email: SinapsisAI <dev@sinapsis.tech>
 Project-URL: Homepage, https://sinapsis.tech

{sinapsis_anomalib-0.1.9 → sinapsis_anomalib-0.1.11}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "sinapsis-anomalib"
-version = "0.1.9"
+version = "0.1.11"
 description = "Templates for anomaly detection with computer vision using anomalib library."
 authors = [
     {name = "SinapsisAI", email = "dev@sinapsis.tech"},

{sinapsis_anomalib-0.1.9 → sinapsis_anomalib-0.1.11}/src/sinapsis_anomalib/helpers/configs.py

@@ -61,7 +61,7 @@ class TrainerConfig(BaseModel):
     devices: int | list[int] | str | None = "auto"
     accelerator: Literal["cpu", "gpu", "tpu", "hpu", "auto"] = "cpu"
     min_epochs: int = 1
-    max_epochs: int | None = None
+    max_epochs: int | None = 5
     model_config = ConfigDict(extra="allow")
 
 

{sinapsis_anomalib-0.1.9 → sinapsis_anomalib-0.1.11}/src/sinapsis_anomalib/templates/anomalib_base.py

@@ -11,7 +11,7 @@ from anomalib.engine import Engine
 from anomalib.utils.types import NORMALIZATION, THRESHOLD
 from lightning.pytorch.callbacks import Callback
 from lightning.pytorch.loggers import Logger
-from pydantic import BaseModel, ConfigDict
+from pydantic import BaseModel, ConfigDict, Field
 from pydantic.dataclasses import dataclass
 from sinapsis_core.data_containers.data_packet import DataContainer
 from sinapsis_core.template_base.base_models import (
@@ -131,7 +131,7 @@ class AnomalibBaseAttributes(TemplateAttributes):
         logger_configs (dict[str, dict[str, Any]] | None): Logger configs
     """
 
-    folder_attributes: FolderConfig | None = None
+    folder_attributes: FolderConfig = Field(default_factory=FolderConfig)
     callbacks: list[Callback] | None = None
     normalization: NORMALIZATION | None = None
     threshold: THRESHOLD | None = None

{sinapsis_anomalib-0.1.9 → sinapsis_anomalib-0.1.11}/src/sinapsis_anomalib/templates/anomalib_export.py

@@ -48,6 +48,7 @@ class AnomalibExportAttributes(AnomalibBaseAttributes):
         generic_key_chkpt (str | None): Key to retrieve training results.
     """
 
+    folder_attributes: dict | None = None
     export_type: ExportType = ExportType.TORCH
     export_root: str | Path | None = None
     input_size: tuple[int, int] | None = None
@@ -153,7 +154,8 @@ class AnomalibExport(AnomalibBase):
         generic_data = self._get_generic_data(container, self.attributes.generic_key_chkpt)
         if generic_data and isinstance(generic_data, AnomalibTrainDataClass):
             return generic_data.checkpoint_path
-
+        if generic_data and isinstance(generic_data, dict):
+            return generic_data.get("checkpoint_path")
         raise ValueError("No checkpoint path found")
 
     def export_model(self, container: DataContainer) -> AnomalibExportDataClass:

{sinapsis_anomalib-0.1.9 → sinapsis_anomalib-0.1.11/src/sinapsis_anomalib.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sinapsis-anomalib
-Version: 0.1.9
+Version: 0.1.11
 Summary: Templates for anomaly detection with computer vision using anomalib library.
 Author-email: SinapsisAI <dev@sinapsis.tech>
 Project-URL: Homepage, https://sinapsis.tech