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

quadra/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "2.3.0a2"
+__version__ = "2.3.1"
 
 
 def get_version():

quadra/callbacks/anomalib.py

@@ -64,7 +64,7 @@ class Visualizer:
         self.figure.subplots_adjust(right=0.9)
 
         axes = self.axis if len(self.images) > 1 else [self.axis]
-        for axis, image_dict in zip(axes, self.images):
+        for axis, image_dict in zip(axes, self.images, strict=False):
             axis.axes.xaxis.set_visible(False)
             axis.axes.yaxis.set_visible(False)
             axis.imshow(image_dict["image"], image_dict["color_map"], vmin=0, vmax=255)
@@ -201,6 +201,7 @@ class VisualizerCallback(Callback):
                 outputs["label"],
                 outputs["pred_labels"],
                 outputs["pred_scores"],
+                strict=False,
             )
         ):
             denormalized_image = Denormalize()(image.cpu())
@@ -256,7 +257,7 @@ class VisualizerCallback(Callback):
             visualizer.close()
 
             if self.plot_raw_outputs:
-                for raw_output, raw_name in zip([heatmap, vis_img], ["heatmap", "segmentation"]):
+                for raw_output, raw_name in zip([heatmap, vis_img], ["heatmap", "segmentation"], strict=False):
                     current_raw_output = raw_output
                     if raw_name == "segmentation":
                         current_raw_output = (raw_output * 255).astype(np.uint8)

quadra/callbacks/lightning.py

@@ -46,7 +46,7 @@ def _scale_batch_size(
             however in practise a few are needed
         init_val: initial batch size to start the search with
         max_trials: max number of increases in batch size done before
-           algorithm is terminated
+            algorithm is terminated
         batch_arg_name: name of the attribute that stores the batch size.
             It is expected that the user has provided a model or datamodule that has a hyperparameter
             with that name. We will look for this attribute name in the following places
@@ -79,6 +79,8 @@ def _scale_batch_size(
         new_size = _run_power_scaling(trainer, init_val, batch_arg_name, max_trials, params)
     elif mode == "binsearch":
         new_size = _run_binary_scaling(trainer, init_val, batch_arg_name, max_trials, params)
+    else:
+        raise ValueError(f"Unknown mode {mode}")
 
     garbage_collection_cuda()
 

quadra/datamodules/base.py

@@ -7,7 +7,7 @@ import pickle as pkl
 import typing
 from collections.abc import Callable, Iterable, Sequence
 from functools import wraps
-from typing import Any, Literal, Union, cast
+from typing import Any, Literal, cast
 
 import albumentations
 import numpy as np
@@ -20,8 +20,8 @@ from tqdm import tqdm
 from quadra.utils import utils
 
 log = utils.get_logger(__name__)
-TrainDataset = Union[torch.utils.data.Dataset, Sequence[torch.utils.data.Dataset]]
-ValDataset = Union[torch.utils.data.Dataset, Sequence[torch.utils.data.Dataset]]
+TrainDataset = torch.utils.data.Dataset | Sequence[torch.utils.data.Dataset]
+ValDataset = torch.utils.data.Dataset | Sequence[torch.utils.data.Dataset]
 TestDataset = torch.utils.data.Dataset
 
 
@@ -260,7 +260,7 @@ class BaseDataModule(LightningDataModule, metaclass=DecorateParentMethod):
             return
 
         # TODO: We need to find a way to annotate the columns of data.
-        paths_and_hash_length = zip(self.data["samples"], [self.hash_size] * len(self.data))
+        paths_and_hash_length = zip(self.data["samples"], [self.hash_size] * len(self.data), strict=False)
 
         with mp.Pool(min(8, mp.cpu_count() - 1)) as pool:
             self.data["hash"] = list(
@@ -355,7 +355,7 @@ class BaseDataModule(LightningDataModule, metaclass=DecorateParentMethod):
             raise ValueError("`n_aug_to_take` is not set. Cannot load augmented samples.")
         aug_samples = []
         aug_labels = []
-        for sample, label in zip(samples, targets):
+        for sample, label in zip(samples, targets, strict=False):
             aug_samples.append(sample)
             aug_labels.append(label)
             final_sample = sample

quadra/datamodules/classification.py

@@ -243,7 +243,7 @@ class ClassificationDataModule(BaseDataModule):
                 samples_test, targets_test = self._read_split(self.test_split_file)
                 if not self.train_split_file:
                     samples_train, targets_train = [], []
-                    for sample, target in zip(all_samples, all_targets):
+                    for sample, target in zip(all_samples, all_targets, strict=False):
                         if sample not in samples_test:
                             samples_train.append(sample)
                             targets_train.append(target)
@@ -251,7 +251,7 @@ class ClassificationDataModule(BaseDataModule):
                 samples_train, targets_train = self._read_split(self.train_split_file)
                 if not self.test_split_file:
                     samples_test, targets_test = [], []
-                    for sample, target in zip(all_samples, all_targets):
+                    for sample, target in zip(all_samples, all_targets, strict=False):
                         if sample not in samples_train:
                             samples_test.append(sample)
                             targets_test.append(target)