quadra 2.2.4a0__py3-none-any.whl → 2.2.5__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (7) hide show
  1. quadra/__init__.py +1 -1
  2. quadra/utils/export.py +6 -1
  3. {quadra-2.2.4a0.dist-info → quadra-2.2.5.dist-info}/METADATA +2 -2
  4. {quadra-2.2.4a0.dist-info → quadra-2.2.5.dist-info}/RECORD +7 -7
  5. {quadra-2.2.4a0.dist-info → quadra-2.2.5.dist-info}/WHEEL +1 -1
  6. {quadra-2.2.4a0.dist-info → quadra-2.2.5.dist-info}/LICENSE +0 -0
  7. {quadra-2.2.4a0.dist-info → quadra-2.2.5.dist-info}/entry_points.txt +0 -0
quadra/__init__.py CHANGED
@@ -1,4 +1,4 @@
1
- __version__ = "2.2.4a0"
1
+ __version__ = "2.2.5"
2
2
 
3
3
 
4
4
  def get_version():
quadra/utils/export.py CHANGED
@@ -1,5 +1,6 @@
1
1
  from __future__ import annotations
2
2
 
3
+ import contextlib
3
4
  import os
4
5
  from collections.abc import Sequence
5
6
  from typing import Any, Literal, TypeVar, cast
@@ -377,7 +378,11 @@ def _safe_export_half_precision_onnx(
377
378
  model_fp32 = onnx.load(export_model_path)
378
379
  test_data = {input_names[i]: inp[i].float().cpu().numpy() for i in range(len(inp))}
379
380
  log.warning("Attempting to convert model in mixed precision, this may take a while...")
380
- model_fp16 = auto_convert_mixed_precision(model_fp32, test_data, rtol=0.01, atol=0.001, keep_io_types=False)
381
+ with open(os.devnull, "w") as f, contextlib.redirect_stdout(f):
382
+ # This function prints a lot of information that is not useful for the user
383
+ model_fp16 = auto_convert_mixed_precision(
384
+ model_fp32, test_data, rtol=0.01, atol=0.001, keep_io_types=False
385
+ )
381
386
  onnx.save(model_fp16, export_model_path)
382
387
 
383
388
  onnx_model = onnx.load(export_model_path)
{quadra-2.2.4a0.dist-info → quadra-2.2.5.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: quadra
3
- Version: 2.2.4a0
3
+ Version: 2.2.5
4
4
  Summary: Deep Learning experiment orchestration library
5
5
  Home-page: https://orobix.github.io/quadra
6
6
  License: Apache-2.0
@@ -21,7 +21,7 @@ Classifier: Topic :: Software Development :: Libraries
21
21
  Classifier: Topic :: Software Development :: Libraries :: Python Modules
22
22
  Provides-Extra: onnx
23
23
  Requires-Dist: albumentations (>=1.3,<1.4)
24
- Requires-Dist: anomalib-orobix (==0.7.0.rc1421)
24
+ Requires-Dist: anomalib-orobix (==0.7.0.dev143)
25
25
  Requires-Dist: boto3 (>=1.26,<1.27)
26
26
  Requires-Dist: grad-cam-orobix (==1.5.3.dev001)
27
27
  Requires-Dist: h5py (>=3.8,<3.9)
{quadra-2.2.4a0.dist-info → quadra-2.2.5.dist-info}/RECORD RENAMED
@@ -1,4 +1,4 @@
1
- quadra/__init__.py,sha256=MrCUud4razb53xIsVpRXHJQ_2EbQA5lL5hRRsoGhFvM,114
1
+ quadra/__init__.py,sha256=wi1UBd-2e2ycj46ChXS4E9NfP8aKtbcXQohx6bZlmsg,112
2
2
  quadra/callbacks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
3
3
  quadra/callbacks/anomalib.py,sha256=Z0Xx3M9UEMYGY0BpSsqemgQSQ6jAr4SrvlbDg_oG9C8,11913
4
4
  quadra/callbacks/lightning.py,sha256=1OTM6fB7qBVLPWNBAZJOb3B00q0kAxMWkPjTEn9YgF0,20182
@@ -262,7 +262,7 @@ quadra/utils/anomaly.py,sha256=49vFvT5-4SxczsEM2Akcut_M1DDwKlOVdGv36oLTgR0,4067
262
262
  quadra/utils/classification.py,sha256=7Phbywpd1CNzwLM5F7zV9FrzdHGd6EkR6LRTRpkerTc,24882
263
263
  quadra/utils/deprecation.py,sha256=zF_S-yqenaZxRBOudhXts0mX763WjEUWCnHd09TZnwY,852
264
264
  quadra/utils/evaluation.py,sha256=5Occ6ONsjnvMkaFFjohtgSvffxEKfjdvMbxqzgtmiHQ,18538
265
- quadra/utils/export.py,sha256=ddTslYMjdm6tYMPZPp-NpLAjUN6jwsgEItTZv2NBx0c,20608
265
+ quadra/utils/export.py,sha256=PoKaADeSwGPIW5TEHP3Klw_SUSV1gP1dIN4CN-AgDm0,20836
266
266
  quadra/utils/imaging.py,sha256=Cz7sGb_axEmnGcwQJP2djFZpIpGCPFIBGT8NWVV-OOE,866
267
267
  quadra/utils/logger.py,sha256=tQJ4xpTAFKx1g-UUm5K1x7zgoP6qoXpcUHQyu0rOr1w,556
268
268
  quadra/utils/mlflow.py,sha256=7E09JShGm2qO7bLb_8srA_RYdVAudxeDYmX7pMdjoVU,3524
@@ -293,8 +293,8 @@ quadra/utils/validator.py,sha256=eFCGr0ss1gYSpsL31JbsCXPZUMJAI9_H-mGodt6UGsU,466
293
293
  quadra/utils/visualization.py,sha256=UvGHX0dumfjpT_KX3Yc1W2B5sAXXaZZWIwXQAi4sdoQ,15950
294
294
  quadra/utils/vit_explainability.py,sha256=hY0awehj6UkyBhnBlW5uWoJTsBfgow5Nll9fAqrzmMo,13337
295
295
  hydra_plugins/quadra_searchpath_plugin.py,sha256=AAn4TzR87zUK7nwSsK-KoqALiPtfQ8FvX3fgZPTGIJ0,1189
296
- quadra-2.2.4a0.dist-info/LICENSE,sha256=8cTbQtcWa02YJoSpMeV_gxj3jpMTkxvl-w3WJ5gV_QE,11342
297
- quadra-2.2.4a0.dist-info/METADATA,sha256=A1R1c1ojPjh_56fQN40J8ryMyEN32XmKamsss2jUOJc,17625
298
- quadra-2.2.4a0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
299
- quadra-2.2.4a0.dist-info/entry_points.txt,sha256=sRYonBZyx-sAJeWcQNQoVQIU5lm02cnCQt6b15k0WHU,43
300
- quadra-2.2.4a0.dist-info/RECORD,,
296
+ quadra-2.2.5.dist-info/LICENSE,sha256=8cTbQtcWa02YJoSpMeV_gxj3jpMTkxvl-w3WJ5gV_QE,11342
297
+ quadra-2.2.5.dist-info/METADATA,sha256=rG202HeIVu_pgeoNRHitD6Gs_dQj_fQcmmN1aDEUYz4,17623
298
+ quadra-2.2.5.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
299
+ quadra-2.2.5.dist-info/entry_points.txt,sha256=sRYonBZyx-sAJeWcQNQoVQIU5lm02cnCQt6b15k0WHU,43
300
+ quadra-2.2.5.dist-info/RECORD,,
{quadra-2.2.4a0.dist-info → quadra-2.2.5.dist-info}/WHEEL RENAMED
@@ -1,4 +1,4 @@
1
1
  Wheel-Version: 1.0
2
- Generator: poetry-core 1.9.0
2
+ Generator: poetry-core 1.9.1
3
3
  Root-Is-Purelib: true
4
4
  Tag: py3-none-any