quadra 2.3.0a2__py3-none-any.whl → 2.3.1__py3-none-any.whl

quadra/utils/export.py

@@ -45,11 +45,11 @@ def generate_torch_inputs(
     """
     inp = None
 
-    if isinstance(input_shapes, (ListConfig, DictConfig)):
+    if isinstance(input_shapes, ListConfig | DictConfig):
         input_shapes = OmegaConf.to_container(input_shapes)
 
     if isinstance(input_shapes, list):
-        if any(isinstance(inp, (Sequence, dict)) for inp in input_shapes):
+        if any(isinstance(inp, Sequence | dict) for inp in input_shapes):
             return [generate_torch_inputs(inp, device, half_precision, dtype) for inp in input_shapes]
 
         # Base case
@@ -59,7 +59,7 @@ def generate_torch_inputs(
         return {k: generate_torch_inputs(v, device, half_precision, dtype) for k, v in input_shapes.items()}
 
     if isinstance(input_shapes, tuple):
-        if any(isinstance(inp, (Sequence, dict)) for inp in input_shapes):
+        if any(isinstance(inp, Sequence | dict) for inp in input_shapes):
             # The tuple contains a list, tuple or dict
             return tuple(generate_torch_inputs(inp, device, half_precision, dtype) for inp in input_shapes)
 
@@ -324,7 +324,7 @@ def _safe_export_half_precision_onnx(
     onnx_config: DictConfig,
     input_shapes: list[Any],
     input_names: list[str],
-):
+) -> bool:
     """Check that the exported half precision ONNX model does not contain NaN values. If it does, attempt to export
     the model with a more stable export and overwrite the original model.
 
@@ -381,7 +381,7 @@ def _safe_export_half_precision_onnx(
             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
+                    model_fp32, test_data, rtol=0.01, atol=5e-3, keep_io_types=False
                 )
             onnx.save(model_fp16, export_model_path)
 

quadra/utils/mlflow.py

@@ -11,6 +11,7 @@ except ImportError:
 from collections.abc import Sequence
 from typing import Any
 
+import numpy as np
 import torch
 from pytorch_lightning import Trainer
 from pytorch_lightning.loggers import MLFlowLogger
@@ -45,6 +46,7 @@ def infer_signature_input(input_tensor: Any) -> Any:
     Raises:
         ValueError: If the input type is not supported or when nested dicts or sequences are encountered.
     """
+    signature: dict[str, Any] | np.ndarray
     if isinstance(input_tensor, Sequence):
         # Mlflow currently does not support sequence outputs, so we use a dict instead
         signature = {}

quadra/utils/models.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 import math
 import warnings
 from collections.abc import Callable
-from typing import Union, cast
+from typing import cast
 
 import numpy as np
 import timm
@@ -114,7 +114,7 @@ def get_feature(
             labels: input_labels
             grayscale_cams: Gradcam output maps, None if gradcam arg is False
     """
-    if isinstance(feature_extractor, (TorchEvaluationModel, TorchscriptEvaluationModel)):
+    if isinstance(feature_extractor, TorchEvaluationModel | TorchscriptEvaluationModel):
         # If we are working with torch based evaluation models we need to extract the model
         feature_extractor = feature_extractor.model
     elif isinstance(feature_extractor, ONNXEvaluationModel):
@@ -160,9 +160,7 @@ def get_feature(
                 x1 = x1.to(feature_extractor.device).to(feature_extractor.model_dtype)
 
             if gradcam:
-                y_hat = cast(
-                    Union[list[torch.Tensor], tuple[torch.Tensor], torch.Tensor], feature_extractor(x1).detach()
-                )
+                y_hat = cast(list[torch.Tensor] | tuple[torch.Tensor] | torch.Tensor, feature_extractor(x1).detach())
                 # mypy can't detect that gradcam is true only if we have a features_extractor
                 if is_vision_transformer(feature_extractor.features_extractor):  # type: ignore[union-attr]
                     grayscale_cam_low_res = grad_rollout(
@@ -177,10 +175,10 @@ def get_feature(
                 feature_extractor.zero_grad(set_to_none=True)  # type: ignore[union-attr]
             else:
                 with torch.no_grad():
-                    y_hat = cast(Union[list[torch.Tensor], tuple[torch.Tensor], torch.Tensor], feature_extractor(x1))
+                    y_hat = cast(list[torch.Tensor] | tuple[torch.Tensor] | torch.Tensor, feature_extractor(x1))
                 grayscale_cams = None
 
-            if isinstance(y_hat, (list, tuple)):
+            if isinstance(y_hat, list | tuple):
                 y_hat = y_hat[0].cpu()
             else:
                 y_hat = y_hat.cpu()

quadra/utils/patch/dataset.py

@@ -566,7 +566,7 @@ def generate_patch_dataset(
             num_workers=num_workers,
         )
 
-    for phase, split_dict in zip(["val", "test"], [val_data_dictionary, test_data_dictionary]):
+    for phase, split_dict in zip(["val", "test"], [val_data_dictionary, test_data_dictionary], strict=False):
         if len(split_dict) > 0:
             log.info("Generating %s set", phase)
             generate_patch_sliding_window_dataset(
@@ -908,9 +908,9 @@ def extract_patches(
             patches = np.concatenate([patches, extra_patches_h], axis=0)
 
     # If this is not true there's some strange case I didn't take into account
-    assert (
-        patches.shape[0] == patch_num_h and patches.shape[1] == patch_num_w
-    ), f"Patch shape {patches.shape} does not match the expected shape {patch_number}"
+    assert patches.shape[0] == patch_num_h and patches.shape[1] == patch_num_w, (
+        f"Patch shape {patches.shape} does not match the expected shape {patch_number}"
+    )
 
     return patches
 
@@ -1059,11 +1059,12 @@ def create_h5(
         h = img.shape[0]
         w = img.shape[1]
 
+        mask: np.ndarray
         if item["mask"] is None:
-            mask = np.zeros([h, w])
+            mask = np.zeros([h, w], dtype=np.uint8)
         else:
             # this works even if item["mask"] is already an absolute path
-            mask = cv2.imread(os.path.join(output_folder, item["mask"]), 0)  # type: ignore[assignment]
+            mask = cv2.imread(os.path.join(output_folder, item["mask"]), 0)
 
         if patch_size is not None:
             patch_height = patch_size[1]

quadra/utils/patch/metrics.py

@@ -98,9 +98,9 @@ def compute_patch_metrics(
     if (patch_h is not None and patch_w is not None) and (patch_num_h is not None and patch_num_w is not None):
         raise ValueError("Either number of patches or patch size is required for reconstruction")
 
-    assert (patch_h is not None and patch_w is not None) or (
-        patch_num_h is not None and patch_num_w is not None
-    ), "Either number of patches or patch size is required for reconstruction"
+    assert (patch_h is not None and patch_w is not None) or (patch_num_h is not None and patch_num_w is not None), (
+        "Either number of patches or patch size is required for reconstruction"
+    )
 
     if patch_h is not None and patch_w is not None and patch_num_h is not None and patch_num_w is not None:
         warnings.warn(
@@ -191,7 +191,7 @@ def compute_patch_metrics(
             if annotated_good is not None:
                 gt_img[np.isin(gt_img, annotated_good)] = 0
 
-            gt_img_binary = (gt_img > 0).astype(bool)  # type: ignore[operator]
+            gt_img_binary = (gt_img > 0).astype(bool)
             regions_pred = label(output_mask).astype(np.uint8)
 
             for k in range(1, regions_pred.max() + 1):
@@ -203,8 +203,11 @@ def compute_patch_metrics(
             output_mask = (output_mask > 0).astype(np.uint8)
             gt_img = label(gt_img)
 
-            for i in range(1, gt_img.max() + 1):  # type: ignore[union-attr]
-                region = (gt_img == i).astype(bool)  # type: ignore[union-attr]
+            if gt_img is None:
+                raise RuntimeError("Ground truth mask is None after label and it should not be")
+
+            for i in range(1, gt_img.max() + 1):
+                region = (gt_img == i).astype(bool)
                 if np.sum(np.bitwise_and(region, output_mask)) == 0:
                     false_region_good += 1
                 else:

quadra/utils/patch/visualization.py

@@ -69,13 +69,13 @@ def plot_patch_reconstruction(
             points = [[item["x"], item["y"]] for item in region["points"]]
             c_label = region["label"]
 
-            out = cv2.drawContours(
+            out = cv2.drawContours(  # type: ignore[call-overload]
                 out,
                 np.array([points], np.int32),
                 -1,
                 class_to_idx[c_label],
                 thickness=cv2.FILLED,
-            )  # type: ignore[call-overload]
+            )
     else:
         out = reconstruction["prediction"]
 

quadra/utils/tests/fixtures/dataset/imagenette.py

@@ -43,7 +43,7 @@ def imagenette_dataset(tmp_path: Path) -> str:
 
     Args:
         tmp_path: Path to temporary directory
-        request: Pytest SubRequest object
+
     Yields:
         Path to imagenette dataset folder
     """

quadra/utils/utils.py

@@ -438,7 +438,7 @@ def flatten_list(input_list: Iterable[Any]) -> Iterator[Any]:
         The iterator over the flattend list
     """
     for v in input_list:
-        if isinstance(v, Iterable) and not isinstance(v, (str, bytes)):
+        if isinstance(v, Iterable) and not isinstance(v, str | bytes):
             yield from flatten_list(v)
         else:
             yield v

quadra/utils/validator.py

@@ -72,9 +72,7 @@ def check_all_arguments(callable_variable: str, configuration_arguments: list[st
     """
     for argument in configuration_arguments:
         if argument not in argument_names:
-            error_string = (
-                f"`{argument}` is not a valid argument passed " f"from configuration to `{callable_variable}`."
-            )
+            error_string = f"`{argument}` is not a valid argument passed from configuration to `{callable_variable}`."
             closest_match = difflib.get_close_matches(argument, argument_names, n=1, cutoff=0.5)
             if len(closest_match) > 0:
                 error_string += f" Did you mean `{closest_match[0]}`?"

quadra/utils/visualization.py

@@ -46,7 +46,7 @@ class UnNormalize:
             new_t = tensor.detach().clone()
         else:
             new_t = tensor
-        for t, m, s in zip(new_t, self.mean, self.std):
+        for t, m, s in zip(new_t, self.mean, self.std, strict=False):
             t.mul_(s).add_(m)
             # The normalize code -> t.sub_(m).div_(s)
         return new_t
@@ -82,7 +82,7 @@ def create_grid_figure(
             ax[i][j].get_xaxis().set_ticks([])
             ax[i][j].get_yaxis().set_ticks([])
     if row_names is not None:
-        for ax, name in zip(ax[:, 0], row_names):  # noqa: B020
+        for ax, name in zip(ax[:, 0], row_names, strict=False):  # noqa: B020
             ax.set_ylabel(name, rotation=90)
 
     plt.tight_layout()
@@ -98,12 +98,12 @@ def create_visualization_dataset(dataset: torch.utils.data.Dataset):
         """Handle different types of transforms."""
         if isinstance(transforms, albumentations.BaseCompose):
             transforms.transforms = convert_transforms(transforms.transforms)
-        if isinstance(transforms, (list, ListConfig, TransformsSeqType)):
+        if isinstance(transforms, list | ListConfig | TransformsSeqType):
             transforms = [convert_transforms(t) for t in transforms]
-        if isinstance(transforms, (dict, DictConfig)):
+        if isinstance(transforms, dict | DictConfig):
             for tname, t in transforms.items():
                 transforms[tname] = convert_transforms(t)
-        if isinstance(transforms, (Normalize, ToTensorV2)):
+        if isinstance(transforms, Normalize | ToTensorV2):
             return NoOp(p=1)
         return transforms
 
@@ -362,6 +362,9 @@ def plot_classification_results(
                         test_label = idx_to_class[test_labels[i]]
                     except Exception:
                         test_label = test_labels[i]
+                else:
+                    pred_label = pred_labels[i]
+                    test_label = test_labels[i]
 
                 ax.axis("off")
                 ax.set_title(f"True: {str(test_label)}\nPred {str(pred_label)}")

quadra/utils/vit_explainability.py

@@ -153,7 +153,7 @@ def grad_rollout(
     """
     result = torch.eye(attentions[0].size(-1))
     with torch.no_grad():
-        for attention, grad in zip(attentions, gradients):
+        for attention, grad in zip(attentions, gradients, strict=False):
             weights = grad
             attention_heads_fused = torch.mean((attention * weights), dim=1)
             attention_heads_fused[attention_heads_fused < 0] = 0

{quadra-2.3.0a2.dist-info → quadra-2.3.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: quadra
-Version: 2.3.0a2
+Version: 2.3.1
 Summary: Deep Learning experiment orchestration library
 Home-page: https://orobix.github.io/quadra
 License: Apache-2.0

{quadra-2.3.0a2.dist-info → quadra-2.3.1.dist-info}/RECORD

@@ -1,7 +1,7 @@
-quadra/__init__.py,sha256=H1Lj8bTwd7z9j_afVDcFqTV6qqbn3GX9RHFirYGp7zQ,114
+quadra/__init__.py,sha256=U4GXsdTR6bVbP904TotXxlsdWqOKNrh_fBTlXM3MR6w,112
 quadra/callbacks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-quadra/callbacks/anomalib.py,sha256=Z0Xx3M9UEMYGY0BpSsqemgQSQ6jAr4SrvlbDg_oG9C8,11913
-quadra/callbacks/lightning.py,sha256=1OTM6fB7qBVLPWNBAZJOb3B00q0kAxMWkPjTEn9YgF0,20182
+quadra/callbacks/anomalib.py,sha256=WLBEGhZA9HoP4Yh9UbbC2GzDOKYTkvU9EY1lkZcV7Fs,11971
+quadra/callbacks/lightning.py,sha256=qvtzDiv8ZUV7K11gKHKWCyo-a9XR_Jm_M-IEicTM1Yo,20242
 quadra/callbacks/mlflow.py,sha256=4LKjrgbRCHP5dOCoDpF7J25gaBgABa0Rof-EA61Iqug,10129
 quadra/callbacks/scheduler.py,sha256=zrglcTUvMO236VchQFtCSlA-XXhc6a3HVWX0uDVQoyc,2656
 quadra/configs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -187,22 +187,22 @@ quadra/configs/transforms/dino.yaml,sha256=NtEbtJPHYkR9DOBPwXR33uvrEdqv8WNyqRlXr
 quadra/configs/transforms/linear_eval.yaml,sha256=fXmJBEwTWQ-QBMNV0mSG9wcrj31YGIV_czcRDczc1ss,488
 quadra/datamodules/__init__.py,sha256=y00iX2YAy6CJzPstKSBNq8_1YsYTRr_sCvqaL-WI7Z8,636
 quadra/datamodules/anomaly.py,sha256=_3FZNSwdMj-ECXlPQDslswtaMn0F1EgzA0q0UH-UgFY,6670
-quadra/datamodules/base.py,sha256=vDKyoZMMzKUn9dSU1hNiCW4Tt7G6xFTxuZn6c8CQxIc,14001
-quadra/datamodules/classification.py,sha256=UfszA_14TmS-9xQLdVLVJAjUT2PG1KLzfOX2w9mFKzg,41594
+quadra/datamodules/base.py,sha256=QGkJ8Lq6hznHvaXjD8mhJhrinrs4ZFlZD3-B5cLU0cQ,14010
+quadra/datamodules/classification.py,sha256=VwQd-zhzJuLgq5Kg1niOY4pnRbO7Sk4B77dWiTFv4do,41622
 quadra/datamodules/generic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 quadra/datamodules/generic/imagenette.py,sha256=3hOb-GmvnKx_hqrSqRcAcf22PjtCQ3CY_-5mlaZSTIM,5564
 quadra/datamodules/generic/mnist.py,sha256=j4xWEWQb1utW3yyozgHD1tP0kOAtLpRsgeIBZ1cIiP0,3425
 quadra/datamodules/generic/mvtec.py,sha256=3Ib8JyY1Eg7wbPL2dXw22YCoy_gsitksofFShLQ9Itw,2700
 quadra/datamodules/generic/oxford_pet.py,sha256=tumWy9TBThvVQZ2JOyghosWJEEsYjyXN6pZMJ9C5dBY,6822
 quadra/datamodules/patch.py,sha256=y7leDt1MyVg0LnqKgWCZ0i6cuVln10fiG4X8EFbl-_Q,7789
-quadra/datamodules/segmentation.py,sha256=eWS_v7H4ea9h05KW1nMpLMJjCGVZEnzBU-vuKlVAdeY,28920
+quadra/datamodules/segmentation.py,sha256=hhfOs7QoYslHYfWfnAgZzSusj2tus8k-h7SBqGNVT8E,29004
 quadra/datamodules/ssl.py,sha256=U63FCdcRJjx4K0RZzkKJfvYJhFpvWTnlBBCtXirn_F4,5709
 quadra/datasets/__init__.py,sha256=nVpqp2ffQ6omqCMB3r1ajcUGgUad0eSkDt-kNWDGblU,669
-quadra/datasets/anomaly.py,sha256=Guxb39aZkne5Qp_pEhjPACdzYSkLdLbu5AUXP0Gf1aE,11944
-quadra/datasets/classification.py,sha256=e-hNFh76aFnvbtkQLq1ljJ9Q3a6ymOmFziNjNK8VM4U,7512
-quadra/datasets/patch.py,sha256=bloUJS_qzzSfVqZaFpQd0uJMhRKRCfEwnSqCJaVtyZQ,4789
+quadra/datasets/anomaly.py,sha256=4rCd2-frgMH3RfQYVFYn5ZXxTKbPOk8GwE-BZIiLwFY,11892
+quadra/datasets/classification.py,sha256=ISKcY2PwD3HNv1JPPbDIJRJWJmu3KR3hlx3HUxlXYpE,7530
+quadra/datasets/patch.py,sha256=imNJONPoREivSZ-6WqYO2zE80PDEr-oCm3rdJuKlWz0,4803
 quadra/datasets/segmentation.py,sha256=cDs45eRh_IBSLB0K5xDos-D4KySRQN64BzaPKGBF7OI,9056
-quadra/datasets/ssl.py,sha256=bbGWM-mQvr5xqXtmanr6HKC1Hgq42asu7wx2prb5NVo,3925
+quadra/datasets/ssl.py,sha256=FLL3dYCKnMymtwZfPEi0TzXI6lh6X3HpbqVzaEoGbeU,3931
 quadra/losses/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 quadra/losses/classification/__init__.py,sha256=R1rhnZsksSrY0Tntc7ITViszbbW6i_705zyLlMpcjPs,153
 quadra/losses/classification/asl.py,sha256=ywfT_ifkHoA7VpAhOiJty0OqzKwFqe0OU5Ands1cI0I,2844
@@ -219,21 +219,21 @@ quadra/losses/ssl/simsiam.py,sha256=uCCbqU9aYMwNa3re0qkeEK5Iz7Hxi0jAcEc-sCWZ8fc,
 quadra/losses/ssl/vicreg.py,sha256=ANvhYJz6iPv1A-OBXgBSrZrDG-1VmPtK1IZDtyFqNHE,2427
 quadra/main.py,sha256=6ZYKytVvCzQjgP_0QA6-3ICzVppsbRgPjF-csLKv85o,1407
 quadra/metrics/__init__.py,sha256=HsTK1gxsjp8_MYgA5caa4OK8sXLqtK_tt9wYyjtFnOc,79
-quadra/metrics/segmentation.py,sha256=jkQzCUz0ibkEoF95-uL-4wwxybs5SAs2VHiT5ZGckso,9451
+quadra/metrics/segmentation.py,sha256=tVRYEyMiwD0RJ7NtoGRoSbwb8sAKoVmvzEhV6-3iQT4,9465
 quadra/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-quadra/models/base.py,sha256=nYbZ1JRmhgcORZhAJoV6d2_L5C5yds_X-PQAFIDoHWk,5514
+quadra/models/base.py,sha256=QqMRQWqIsUuUHeInqMHZv3wv7Xeqz-zKe4cAQeqaN3M,5544
 quadra/models/classification/__init__.py,sha256=c03CGDyvipXgU7ybSosOaeTl1aM1ge6TqMUgMiTpQtA,243
 quadra/models/classification/backbones.py,sha256=haHNPC-XZ8Jj1i47cfUj8JHy_I-rins-nNfccrPBffo,6281
 quadra/models/classification/base.py,sha256=w-mDPQPtIrNclxjqsve5BTmNhNgnWGh7uJfE5HaTFPA,2996
-quadra/models/evaluation.py,sha256=dEazagyUvlnnEq16IdM25qAdDXZR-BUhApDF_YUdO-E,10695
+quadra/models/evaluation.py,sha256=LQg2K6PDIKK0ZnkP4pHfRNnKO4WeaROoYoNFA3Bctg0,10709
 quadra/modules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 quadra/modules/backbone.py,sha256=xiZBqgzr1S45GX9mydl29TFuahLDaHrU7Fy73LGIyGI,909
-quadra/modules/base.py,sha256=fS_kbOAfEihHFfiTcyuyDXTzkmkJr2ivtAMiZqjOUog,10303
+quadra/modules/base.py,sha256=y96PSFJeo4gswVj3a6uNnoirg-dMgS0MsYDN51fQQ9A,10382
 quadra/modules/classification/__init__.py,sha256=6keltBhC1yzgbNttBuykNYJAUMyOrY-HDNgGZGfI93I,141
 quadra/modules/classification/base.py,sha256=QdHtHY2tF_qh2wU01Oo0TWjh9CTqa46tyF4VgcLd__M,11937
 quadra/modules/ssl/__init__.py,sha256=oeUoGHrsESZ0595-JxPxURBP124jtNfrITbVovBpANA,302
 quadra/modules/ssl/barlowtwins.py,sha256=iW6f7ADSEkbs7z-88x680204-Ez-iF1Yd2SdQzcLpRY,1884
-quadra/modules/ssl/byol.py,sha256=8sviU7MS9MXDbS9Ogu9-qJCZrkTI2pPxgEwNvc7EqIo,7084
+quadra/modules/ssl/byol.py,sha256=3UhUr72kpI2lM9JtVPqrTcpTo60NsAHNu3SIwD5_RrI,7114
 quadra/modules/ssl/common.py,sha256=nQMsYEu4PUueMq0KNe898h3wGS2RVQBN0NCpYnMyRqI,9898
 quadra/modules/ssl/dino.py,sha256=Xs4wRYvvxeLuHtOW5Gf-xaqAvT97cIuOG6PlYduPDm4,7300
 quadra/modules/ssl/hyperspherical.py,sha256=yEY0WvYFLvKCeKKJDAWCEttYwNVjB5ai6N2FxXKqYQ4,6356
@@ -248,31 +248,31 @@ quadra/schedulers/__init__.py,sha256=mQivr18c0j36hpV3Lm8nlyBVKFevWp8TtLuTfvI9kQc
 quadra/schedulers/base.py,sha256=T1EdrLOJ0i9MzWoLCkrNA0uypm7hJ-L6NFhjIXFB6NE,1462
 quadra/schedulers/warmup.py,sha256=chzzrK7OqqlicBCxiF4CqMYNrWu6nflIbRE-C86Jrw0,4962
 quadra/tasks/__init__.py,sha256=tmAfMoH0k3UC7r2pNrgbBa1Pfc3tpLl3IObFF6Z0eRE,820
-quadra/tasks/anomaly.py,sha256=rl3F39kABt5deSLOeY2s_1t2mR4x1-42VdP0flRrZMs,24583
-quadra/tasks/base.py,sha256=5Rsjdothqb4YXMIN_s98HdoPmJBfQKV0ZwnyvUuihYI,14101
-quadra/tasks/classification.py,sha256=I4NEsNVn1lOu-HLaD_hqowfnedhD4l-wZ4JPI1eD4pg,52735
-quadra/tasks/patch.py,sha256=EJvbtvlebyOtk6m47juW8XMXr1v_bLFTIKqI1KC0HRA,20244
-quadra/tasks/segmentation.py,sha256=5GF7CZjm1dQGQ-Q0kAc68GSsNv6FTxBsl_rbEZQwjmU,16213
-quadra/tasks/ssl.py,sha256=SVZeAW5xVkfeGTwMqRhGKLagMC2kHrxoOaNEDzbCZ0A,20552
+quadra/tasks/anomaly.py,sha256=RHeiM1vZF1zsva37iYdiGx_HLgdAp8lXnmUzXja69YU,24638
+quadra/tasks/base.py,sha256=piYlTFtvqH-4s4oEq4GczdAs_gL29UHAJGsOC5Sd3Bc,14187
+quadra/tasks/classification.py,sha256=05l3QM3dsU2yTWhXxNAcJ8sZM0Vbfgey-e5EV6p1TX8,52816
+quadra/tasks/patch.py,sha256=nzo8o-ei7iF1Iarvd8-c08s0Rs_lPvVPDLAbkFMx-Qw,20251
+quadra/tasks/segmentation.py,sha256=9Qy-V0Wvoofl7IrfotnSMgBIXcZd-WfZZtetyqmB0FY,16260
+quadra/tasks/ssl.py,sha256=XsaC9hbhvTA5UfHeRaaCstx9mTYacLRmgoCF5Tj9R5M,20547
 quadra/trainers/README.md,sha256=XtpbUOxwvPpOUL7E5s2JHjRgwT-CRKTxsBeUSXrg9BU,248
 quadra/trainers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 quadra/trainers/classification.py,sha256=YeJ0z7Vk0-dsMTcoKBxSdSA0rxtilEcQTp-Zq9Xi1hw,7042
 quadra/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 quadra/utils/anomaly.py,sha256=49vFvT5-4SxczsEM2Akcut_M1DDwKlOVdGv36oLTgR0,4067
-quadra/utils/classification.py,sha256=7Phbywpd1CNzwLM5F7zV9FrzdHGd6EkR6LRTRpkerTc,24882
+quadra/utils/classification.py,sha256=dKFuv4RywWhvhstOnEOnaf-6qcViUK0dTgah9m9mw2Q,24917
 quadra/utils/deprecation.py,sha256=zF_S-yqenaZxRBOudhXts0mX763WjEUWCnHd09TZnwY,852
-quadra/utils/evaluation.py,sha256=5Occ6ONsjnvMkaFFjohtgSvffxEKfjdvMbxqzgtmiHQ,18538
-quadra/utils/export.py,sha256=PoKaADeSwGPIW5TEHP3Klw_SUSV1gP1dIN4CN-AgDm0,20836
+quadra/utils/evaluation.py,sha256=oooRJPu1AaHhOwvB1Y6SFjQ645OkgrDzKtUvwWq8oq4,19005
+quadra/utils/export.py,sha256=ghNF8mQw-JjZiVeBJ0y8yIQkx8EG8ssPorn3aaIsgcA,20840
 quadra/utils/imaging.py,sha256=Cz7sGb_axEmnGcwQJP2djFZpIpGCPFIBGT8NWVV-OOE,866
 quadra/utils/logger.py,sha256=tQJ4xpTAFKx1g-UUm5K1x7zgoP6qoXpcUHQyu0rOr1w,556
-quadra/utils/mlflow.py,sha256=7E09JShGm2qO7bLb_8srA_RYdVAudxeDYmX7pMdjoVU,3524
+quadra/utils/mlflow.py,sha256=DVso1lxn126hil8i4tTf5WFUPJ8uJNAzNU8OXbXwOzw,3586
 quadra/utils/model_manager.py,sha256=P5JtY95p6giQ6mb4TUnWsNwUh5ClzHBillnG5SA56QY,12546
-quadra/utils/models.py,sha256=xLOT6Sorpc54dYn9XG8RpqoLupvNrMCiE_QvvIMLwHA,19756
+quadra/utils/models.py,sha256=49AXecNN7mg8uqO-YW0sLbPxbvWfTI4E4NNpTesW6HE,19699
 quadra/utils/patch/__init__.py,sha256=YenDdsI937kyAJiE0dP3_Xua8gHIoFjheoWMnpx_TGU,509
-quadra/utils/patch/dataset.py,sha256=hqM7XyPNDmI9_uJSrAxNYOasdNlEHcg-npP1S9Bb05Y,61374
-quadra/utils/patch/metrics.py,sha256=E1PeHFp10pPgkb6484fDvRLn2E9NDy9xIqEmMBeusOw,17644
+quadra/utils/patch/dataset.py,sha256=tRwrc01p0sj4nLQ-6b9mvnkTQrjtFSv5qMYiTJRSXKU,61401
+quadra/utils/patch/metrics.py,sha256=r7zxGXC2hU6EiMbfNoUmi6BC0EEUZs9Jy_mtI5Q1x5g,17693
 quadra/utils/patch/model.py,sha256=F-wbMZvM8nS_ZSYewg2SofD7H0I6DH1DBA2ACSr0fCY,5746
-quadra/utils/patch/visualization.py,sha256=HwO67vRMvhQAKN_FRFv4AAqvYMeOpg0dod3DwGpgTfo,7044
+quadra/utils/patch/visualization.py,sha256=V64SsXcQ2UhBVH2gzzrjF_OaxL58ktEo1Jdzcos3AT8,7044
 quadra/utils/resolver.py,sha256=p8t95b__htcR3hdnF9RtlWNKLTVUWYjADozYNj9lIzQ,1397
 quadra/utils/segmentation.py,sha256=rWOE1qw2RS0dpgJyHqfQURw86K6G2Hst6mpu97PI5Ac,920
 quadra/utils/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -280,7 +280,7 @@ quadra/utils/tests/fixtures/__init__.py,sha256=5KHMpKGK31PRWfdaiM5gmcs3gyTcCQQEv
 quadra/utils/tests/fixtures/dataset/__init__.py,sha256=hfNtCxu3PPAzqKcKw40BnsIsUHNDlGjYhLuQ5n48deA,1298
 quadra/utils/tests/fixtures/dataset/anomaly.py,sha256=x9X_zafn7Kvjya0_ztCwYRXVEoubND_9yoHxYgn9pkY,3899
 quadra/utils/tests/fixtures/dataset/classification.py,sha256=Mh4t1fTS3thd0hdfh-jWD04WKCIBTeU2x1Ds5pJnZmM,15422
-quadra/utils/tests/fixtures/dataset/imagenette.py,sha256=gADjmgiyQNA2CrtjCITlJTqgQG2StL3bUVnADOFSZfQ,1501
+quadra/utils/tests/fixtures/dataset/imagenette.py,sha256=ICwgvYWngEfjaU2JP3riaHHf0FXlOOEyeUtpJ1fYEds,1460
 quadra/utils/tests/fixtures/dataset/segmentation.py,sha256=mSe93hJEpjDdFM69yGoYUzGa-M5T2P8XE4_Z1ZiORuM,5855
 quadra/utils/tests/fixtures/models/__init__.py,sha256=5cxfDtbV-_prBlVu9L2tC34C8QGyng9chUbJYGFv0J0,123
 quadra/utils/tests/fixtures/models/anomaly.py,sha256=J5dG95RlWNds5xqArQY4JlpmqUcLgoESxVC7K5O7ez4,2942
@@ -288,13 +288,13 @@ quadra/utils/tests/fixtures/models/classification.py,sha256=5qpyOonqK6W2LCUWEHhm
 quadra/utils/tests/fixtures/models/segmentation.py,sha256=CTNXeEPcFxFq-YcNfQi5DbbytPZwBQaZn5dQq3L41j0,765
 quadra/utils/tests/helpers.py,sha256=9PJlwozUl_lpQW-Ck-tN7sGFcgeieEd3q56aYuwMIlk,2381
 quadra/utils/tests/models.py,sha256=KbAlv_ukxaUYsyVNUO_dM0NyIosx8RpC0EVyF1HvPkM,507
-quadra/utils/utils.py,sha256=AjVsJvqACj5Bo8aOF8Itkwqhzws3bwfLnptxosbgzl4,19125
-quadra/utils/validator.py,sha256=eFCGr0ss1gYSpsL31JbsCXPZUMJAI9_H-mGodt6UGsU,4668
-quadra/utils/visualization.py,sha256=UvGHX0dumfjpT_KX3Yc1W2B5sAXXaZZWIwXQAi4sdoQ,15950
-quadra/utils/vit_explainability.py,sha256=hY0awehj6UkyBhnBlW5uWoJTsBfgow5Nll9fAqrzmMo,13337
+quadra/utils/utils.py,sha256=3tgj_tFFhKsGNJ9jrmULI9rWxFyhuUe53Y5SBJFkwSM,19124
+quadra/utils/validator.py,sha256=wmVXycB90VNyAbKBUVncFCxK4nsYiOWJIY3ISXwxYCY,4632
+quadra/utils/visualization.py,sha256=yYm7lPziUOlybxigZ2qTycNewb67Q80H4hjQGWUh788,16094
+quadra/utils/vit_explainability.py,sha256=Gh6BHaDEzWxOjJp1aqvCxLt9Rb8TXd5uKXOAx7-acUk,13351
 hydra_plugins/quadra_searchpath_plugin.py,sha256=AAn4TzR87zUK7nwSsK-KoqALiPtfQ8FvX3fgZPTGIJ0,1189
-quadra-2.3.0a2.dist-info/LICENSE,sha256=8cTbQtcWa02YJoSpMeV_gxj3jpMTkxvl-w3WJ5gV_QE,11342
-quadra-2.3.0a2.dist-info/METADATA,sha256=6sS_JtaCenmsM14a9s7_R8LadWrppBFXz5QaNOL_hDA,17600
-quadra-2.3.0a2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-quadra-2.3.0a2.dist-info/entry_points.txt,sha256=sRYonBZyx-sAJeWcQNQoVQIU5lm02cnCQt6b15k0WHU,43
-quadra-2.3.0a2.dist-info/RECORD,,
+quadra-2.3.1.dist-info/LICENSE,sha256=8cTbQtcWa02YJoSpMeV_gxj3jpMTkxvl-w3WJ5gV_QE,11342
+quadra-2.3.1.dist-info/METADATA,sha256=A3ykpI_-NCzmuMhhkGt7Z2Qs6qJT_xz9w4nOfwoZnJg,17598
+quadra-2.3.1.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+quadra-2.3.1.dist-info/entry_points.txt,sha256=sRYonBZyx-sAJeWcQNQoVQIU5lm02cnCQt6b15k0WHU,43
+quadra-2.3.1.dist-info/RECORD,,

{quadra-2.3.0a2.dist-info → quadra-2.3.1.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 1.9.0
+Generator: poetry-core 1.9.1
 Root-Is-Purelib: true
 Tag: py3-none-any