quadra 2.3.0a2__tar.gz → 2.3.1__tar.gz

{quadra-2.3.0a2 → quadra-2.3.1}/PKG-INFO

@@ -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 → quadra-2.3.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "quadra"
-version = "2.3.0a2"
+version = "2.3.1"
 description = "Deep Learning experiment orchestration library"
 authors = [
 	"Federico Belotti <federico.belotti@orobix.com>",
@@ -98,7 +98,7 @@ hydra-plugins = { path = "quadra_hydra_plugin" }
 # Dev dependencies
 interrogate = "~1.5"
 pre_commit = "^3.0"
-pylint = "~2.16"
+pylint = "^3.3"
 types_pyyaml = "~6.0.12"
 mypy = "^1.9.0"
 pandas_stubs = "~1.5.3"
@@ -123,14 +123,14 @@ pytest_env = "~1.1"
 optional = true
 
 [tool.poetry.group.docs.dependencies]
-mkdocs = "1.5.2"
+mkdocs = "~1.6"
 mkdocs_literate_nav = "0.6.0"
 mkdocs_section_index = "0.3.6"
-mkdocstrings = "0.23.0"
-mkdocs_autorefs = "0.5.0"
+mkdocstrings = "~0.28"
+mkdocs_autorefs = "~1.4"
 mkdocs_gen_files = "0.5.0"
 mkdocs_material = "9.2.8"
-mkdocstrings_python = "1.6.2"
+mkdocstrings_python = "~1.16"
 mkdocs_material_extensions = "1.1.1"
 mike = "1.1.2"
 cairosvg = "2.7.0"
@@ -228,7 +228,7 @@ exclude = ["quadra/utils/tests", "tests"]
 
 [tool.ruff]
 extend-include = ["*.ipynb"]
-target-version = "py39"
+target-version = "py310"
 # Orobix guidelines
 line-length = 120
 indent-width = 4

{quadra-2.3.0a2 → quadra-2.3.1}/quadra/__init__.py

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

{quadra-2.3.0a2 → quadra-2.3.1}/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-2.3.0a2 → quadra-2.3.1}/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()