sinapsis-anomalib 0.1.9__py3-none-any.whl → 0.1.11__py3-none-any.whl

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/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/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.dist-info → sinapsis_anomalib-0.1.11.dist-info}/METADATA

@@ -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.dist-info → sinapsis_anomalib-0.1.11.dist-info}/RECORD

@@ -1,17 +1,17 @@
 sinapsis_anomalib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sinapsis_anomalib/helpers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 sinapsis_anomalib/helpers/config_factory.py,sha256=M7CPS2qHgUWBUTYkuVGhaa2S6IJZ8C3zGPulVjgXr_U,4493
-sinapsis_anomalib/helpers/configs.py,sha256=A1Ap7kGCoQhmE72YonmPxp92tkynGdIHeca2UJd6cFE,3472
+sinapsis_anomalib/helpers/configs.py,sha256=LEcZjYWD00RnI_dpzP_AkC1tRewEkY-gxrSXpCghpxA,3469
 sinapsis_anomalib/helpers/tags.py,sha256=Oj8vSh8AbnOfDY4UadOOteQOhjKwXSB3Rv82hpeP2YE,325
 sinapsis_anomalib/templates/__init__.py,sha256=ltv45AB2oIhPw0jRrL0S6HMYYQp63Cr-DBjD0HjK1i0,904
-sinapsis_anomalib/templates/anomalib_base.py,sha256=SbmRQc8NCq3LXaDbcA-qpqwyBW4UxYPvSaS9kytN-sc,8501
+sinapsis_anomalib/templates/anomalib_base.py,sha256=SBj4-9cM_6uh5HbU_iASGzHAtXOBK7xkZklZ0y8MQss,8532
 sinapsis_anomalib/templates/anomalib_base_inference.py,sha256=YxYpiRK8CwzLjYJpSnNZkUq6OE9zSAbYnqNStPV9RBI,8996
-sinapsis_anomalib/templates/anomalib_export.py,sha256=vHUucJO6E9P7e-3okX8WdRv_otN0P6Fa0GKeWc5fA_0,7840
+sinapsis_anomalib/templates/anomalib_export.py,sha256=vpuu-Rfg0L3zD-ssOKgyKQVsmAwKSs8iAbYcxvSTMSU,7996
 sinapsis_anomalib/templates/anomalib_openvino_inference.py,sha256=FPr2iwC3JbasJnlOgdWzO5ugbQj3Kan3xp-oxJGkC1Q,3207
 sinapsis_anomalib/templates/anomalib_torch_inference.py,sha256=Q6aqspIjFpI7WY90jcnqNmqalhYIm_Um2As9Sx1AOeU,2862
 sinapsis_anomalib/templates/anomalib_train.py,sha256=nS0oLtB5lk-RwKTtuH1bWZc4YM3m6LTl_qowc1wBHs4,7424
-sinapsis_anomalib-0.1.9.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
-sinapsis_anomalib-0.1.9.dist-info/METADATA,sha256=FpA0UMcMGzMr02ebn-EaJqsQ8P8MjpvIZdI4jIkyKLs,7552
-sinapsis_anomalib-0.1.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-sinapsis_anomalib-0.1.9.dist-info/top_level.txt,sha256=p9ltOdENM_b-Ekvb0AvXmvZRV4Bwtly7c7qeOokACvE,18
-sinapsis_anomalib-0.1.9.dist-info/RECORD,,
+sinapsis_anomalib-0.1.11.dist-info/licenses/LICENSE,sha256=hIahDEOTzuHCU5J2nd07LWwkLW7Hko4UFO__ffsvB-8,34523
+sinapsis_anomalib-0.1.11.dist-info/METADATA,sha256=AUU-2fva58pQZus5gzKQMrGBw7_khPUS2DdTL3HZTR8,7553
+sinapsis_anomalib-0.1.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+sinapsis_anomalib-0.1.11.dist-info/top_level.txt,sha256=p9ltOdENM_b-Ekvb0AvXmvZRV4Bwtly7c7qeOokACvE,18
+sinapsis_anomalib-0.1.11.dist-info/RECORD,,