quadra 2.2.0__tar.gz → 2.2.2__tar.gz

{quadra-2.2.0 → quadra-2.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: quadra
-Version: 2.2.0
+Version: 2.2.2
 Summary: Deep Learning experiment orchestration library
 Home-page: https://orobix.github.io/quadra
 License: Apache-2.0
@@ -21,7 +21,7 @@ Classifier: Topic :: Software Development :: Libraries
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Provides-Extra: onnx
 Requires-Dist: albumentations (>=1.3,<1.4)
-Requires-Dist: anomalib-orobix (==0.7.0.dev134)
+Requires-Dist: anomalib-orobix (==0.7.0.dev140)
 Requires-Dist: boto3 (>=1.26,<1.27)
 Requires-Dist: grad_cam (==1.4.6)
 Requires-Dist: h5py (>=3.8,<3.9)
@@ -110,7 +110,7 @@ If you use pip to manage your packages, you can install `quadra` from PyPi by ru
 pip install quadra
 ```
 
-If instead you prefer to use poetry, you can install `quadra` from source by running the following command:
+If instead you prefer to use poetry, you can install `quadra` from PyPi by running the following command:
 ```shell
 poetry add quadra
 ```
@@ -235,7 +235,7 @@ On the same dataset we can run a simple SSL training using the BYOL algorithm wi
 quadra experiment=generic/imagenette/ssl/byol logger=csv
 ```
 
-BYOL is not the only SSL algorithm available, you can find a list of all the available algorithms under `quadra/experiment/generic/imagenette/ssl` folder.
+BYOL is not the only SSL algorithm available, you can find a list of all the available algorithms under `quadra/configs/experiment/generic/imagenette/ssl` folder.
 
 
 ### Anomaly Detection Training
@@ -248,11 +248,11 @@ quadra experiment=generic/mnist/anomaly/padim logger=csv
 
 This will run an anomaly detection considering on of the classes as good (default is the number 9) and the rest as anomalies.
 
-This will make use of the [anomalib](https://github.com/openvinotoolkit/anomalib) library to train the model. Many different algorithms are available, you can find them under `quadra/experiment/generic/mnist/anomaly` folder.
+This will make use of the [anomalib](https://github.com/openvinotoolkit/anomalib) library to train the model. Many different algorithms are available, you can find them under `quadra/configs/experiment/generic/mnist/anomaly` folder.
 
 ## Running with Custom Datasets
 
-Each task comes with a default configuration file that can be customized for your needs. Each example experiment we have seen so far uses a default configuration file that can be found under `quadra/experiment/base/<task>/<config_name>.yaml`. 
+Each task comes with a default configuration file that can be customized for your needs. Each example experiment we have seen so far uses a default configuration file that can be found under `quadra/configs/experiment/base/<task>/<config_name>.yaml`. 
 
 Let's see how we can customize the configuration file to run the classification experiment on a custom dataset.
 

{quadra-2.2.0 → quadra-2.2.2}/README.md

@@ -47,7 +47,7 @@ If you use pip to manage your packages, you can install `quadra` from PyPi by ru
 pip install quadra
 ```
 
-If instead you prefer to use poetry, you can install `quadra` from source by running the following command:
+If instead you prefer to use poetry, you can install `quadra` from PyPi by running the following command:
 ```shell
 poetry add quadra
 ```
@@ -172,7 +172,7 @@ On the same dataset we can run a simple SSL training using the BYOL algorithm wi
 quadra experiment=generic/imagenette/ssl/byol logger=csv
 ```
 
-BYOL is not the only SSL algorithm available, you can find a list of all the available algorithms under `quadra/experiment/generic/imagenette/ssl` folder.
+BYOL is not the only SSL algorithm available, you can find a list of all the available algorithms under `quadra/configs/experiment/generic/imagenette/ssl` folder.
 
 
 ### Anomaly Detection Training
@@ -185,11 +185,11 @@ quadra experiment=generic/mnist/anomaly/padim logger=csv
 
 This will run an anomaly detection considering on of the classes as good (default is the number 9) and the rest as anomalies.
 
-This will make use of the [anomalib](https://github.com/openvinotoolkit/anomalib) library to train the model. Many different algorithms are available, you can find them under `quadra/experiment/generic/mnist/anomaly` folder.
+This will make use of the [anomalib](https://github.com/openvinotoolkit/anomalib) library to train the model. Many different algorithms are available, you can find them under `quadra/configs/experiment/generic/mnist/anomaly` folder.
 
 ## Running with Custom Datasets
 
-Each task comes with a default configuration file that can be customized for your needs. Each example experiment we have seen so far uses a default configuration file that can be found under `quadra/experiment/base/<task>/<config_name>.yaml`. 
+Each task comes with a default configuration file that can be customized for your needs. Each example experiment we have seen so far uses a default configuration file that can be found under `quadra/configs/experiment/base/<task>/<config_name>.yaml`. 
 
 Let's see how we can customize the configuration file to run the classification experiment on a custom dataset.
 

{quadra-2.2.0 → quadra-2.2.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "quadra"
-version = "2.2.0"
+version = "2.2.2"
 description = "Deep Learning experiment orchestration library"
 authors = [
 	"Federico Belotti <federico.belotti@orobix.com>",
@@ -72,7 +72,7 @@ h5py = "~3.8"
 timm = "0.9.12"
 
 segmentation_models_pytorch-orobix = "0.3.3.dev1"
-anomalib-orobix = "0.7.0.dev134"
+anomalib-orobix = "0.7.0.dev140"
 xxhash = "~3.2"
 torchinfo = "~1.8"
 typing_extensions = { version = "4.11.0", python = "<3.10" }

{quadra-2.2.0 → quadra-2.2.2}/quadra/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "2.2.0"
+__version__ = "2.2.2"
 
 
 def get_version():

{quadra-2.2.0 → quadra-2.2.2}/quadra/main.py

@@ -1,6 +1,7 @@
 import time
 
 import hydra
+import matplotlib
 from omegaconf import DictConfig
 from pytorch_lightning import seed_everything
 
@@ -12,6 +13,8 @@ from quadra.utils.validator import validate_config
 load_envs()
 register_resolvers()
 
+
+matplotlib.use("Agg")
 log = get_logger(__name__)
 
 

{quadra-2.2.0 → quadra-2.2.2}/quadra/utils/export.py

@@ -7,7 +7,6 @@ from typing import Any, Literal, TypeVar, cast
 import torch
 from anomalib.models.cflow import CflowLightning
 from omegaconf import DictConfig, ListConfig, OmegaConf
-from onnxconverter_common import auto_convert_mixed_precision
 from torch import nn
 
 from quadra.models.base import ModelSignatureWrapper
@@ -22,6 +21,7 @@ from quadra.utils.logger import get_logger
 try:
     import onnx  # noqa
     from onnxsim import simplify as onnx_simplify  # noqa
+    from onnxconverter_common import auto_convert_mixed_precision  # noqa
 
     ONNX_AVAILABLE = True
 except ImportError:
@@ -385,8 +385,9 @@ def _safe_export_half_precision_onnx(
             onnx.checker.check_model(onnx_model)
             return True
         except Exception as e:
-            log.debug("Failed to export model with mixed precision with error: %s", e)
-            return False
+            raise RuntimeError(
+                "Failed to export model with automatic mixed precision, check your model or disable ONNX export"
+            ) from e
     else:
         log.info("Exported half precision ONNX model does not contain NaN values, model is stable")
         return True