quadra 2.2.4a0__tar.gz → 2.2.5__tar.gz

{quadra-2.2.4a0 → quadra-2.2.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: quadra
-Version: 2.2.4a0
+Version: 2.2.5
 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.rc1421)
+Requires-Dist: anomalib-orobix (==0.7.0.dev143)
 Requires-Dist: boto3 (>=1.26,<1.27)
 Requires-Dist: grad-cam-orobix (==1.5.3.dev001)
 Requires-Dist: h5py (>=3.8,<3.9)

{quadra-2.2.4a0 → quadra-2.2.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "quadra"
-version = "2.2.4a0"
+version = "2.2.5"
 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.rc1421"
+anomalib-orobix = "0.7.0.dev143"
 xxhash = "~3.2"
 torchinfo = "~1.8"
 typing_extensions = { version = "4.11.0", python = "<3.10" }

{quadra-2.2.4a0 → quadra-2.2.5}/quadra/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "2.2.4a0"
+__version__ = "2.2.5"
 
 
 def get_version():

{quadra-2.2.4a0 → quadra-2.2.5}/quadra/utils/export.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import contextlib
 import os
 from collections.abc import Sequence
 from typing import Any, Literal, TypeVar, cast
@@ -377,7 +378,11 @@ def _safe_export_half_precision_onnx(
             model_fp32 = onnx.load(export_model_path)
             test_data = {input_names[i]: inp[i].float().cpu().numpy() for i in range(len(inp))}
             log.warning("Attempting to convert model in mixed precision, this may take a while...")
-            model_fp16 = auto_convert_mixed_precision(model_fp32, test_data, rtol=0.01, atol=0.001, keep_io_types=False)
+            with open(os.devnull, "w") as f, contextlib.redirect_stdout(f):
+                # This function prints a lot of information that is not useful for the user
+                model_fp16 = auto_convert_mixed_precision(
+                    model_fp32, test_data, rtol=0.01, atol=0.001, keep_io_types=False
+                )
             onnx.save(model_fp16, export_model_path)
 
             onnx_model = onnx.load(export_model_path)