quadra 2.6.0__py3-none-any.whl → 2.6.1__py3-none-any.whl

quadra/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "2.6.0"
+__version__ = "2.6.1"
 
 
 def get_version():

quadra/trainers/classification.py

@@ -163,6 +163,9 @@ class SklearnClassificationTrainer:
             }
         )
 
+        if predict_proba:
+            res["probability"] = test_probability
+
         if not all(t == -1 for t in filtered_test_labels):
             test_real_label_cm = np.array(filtered_test_labels)
             if cams is not None:
@@ -171,9 +174,6 @@ class SklearnClassificationTrainer:
             test_real_label_cm = test_real_label_cm[test_real_label_cm != -1].astype(pred_labels_cm.dtype)
             cl_rep, pd_cm, accuracy = get_results(test_real_label_cm, pred_labels_cm, idx_to_class)
 
-            if predict_proba:
-                res["probability"] = test_probability
-
             return cl_rep, pd_cm, accuracy, res, cams
 
         return None, None, None, res, cams

{quadra-2.6.0.dist-info → quadra-2.6.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: quadra
-Version: 2.6.0
+Version: 2.6.1
 Summary: Deep Learning experiment orchestration library
 License: Apache-2.0
 License-File: LICENSE

{quadra-2.6.0.dist-info → quadra-2.6.1.dist-info}/RECORD

@@ -1,4 +1,4 @@
-quadra/__init__.py,sha256=ylxTD-7xItBBl8y254g9lUc-ehYh4LCjfEoDhXhBfUw,112
+quadra/__init__.py,sha256=B5omgSA1INDOjH9Z_sU0BQr2az85t0bNF8u11EZNyuM,112
 quadra/callbacks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 quadra/callbacks/anomalib.py,sha256=WLBEGhZA9HoP4Yh9UbbC2GzDOKYTkvU9EY1lkZcV7Fs,11971
 quadra/callbacks/lightning.py,sha256=qvtzDiv8ZUV7K11gKHKWCyo-a9XR_Jm_M-IEicTM1Yo,20242
@@ -256,7 +256,7 @@ quadra/tasks/segmentation.py,sha256=9Qy-V0Wvoofl7IrfotnSMgBIXcZd-WfZZtetyqmB0FY,
 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/trainers/classification.py,sha256=UBUvRmRwGgaaDjnwoSX9BeoNhZO-7agp3C1a2x1t_Ik,7034
 quadra/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 quadra/utils/anomaly.py,sha256=uyzrFTz5QFTyIEbOT81A6OxXbHceGeiGP-JP3qr65D8,6044
 quadra/utils/classification.py,sha256=dKFuv4RywWhvhstOnEOnaf-6qcViUK0dTgah9m9mw2Q,24917
@@ -293,8 +293,8 @@ quadra/utils/validator.py,sha256=wmVXycB90VNyAbKBUVncFCxK4nsYiOWJIY3ISXwxYCY,463
 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.6.0.dist-info/METADATA,sha256=eAbmWIkIwEc5poNJHwM_9gEu4gH6uXmGlDv2HOfO7iA,17624
-quadra-2.6.0.dist-info/WHEEL,sha256=3ny-bZhpXrU6vSQ1UPG34FoxZBp3lVcvK0LkgUz6VLk,88
-quadra-2.6.0.dist-info/entry_points.txt,sha256=sRYonBZyx-sAJeWcQNQoVQIU5lm02cnCQt6b15k0WHU,43
-quadra-2.6.0.dist-info/licenses/LICENSE,sha256=8cTbQtcWa02YJoSpMeV_gxj3jpMTkxvl-w3WJ5gV_QE,11342
-quadra-2.6.0.dist-info/RECORD,,
+quadra-2.6.1.dist-info/METADATA,sha256=woiJHRB_oIiaqoaRTLaJJFe-nnBfkH8wCUNLByaQXgM,17624
+quadra-2.6.1.dist-info/WHEEL,sha256=3ny-bZhpXrU6vSQ1UPG34FoxZBp3lVcvK0LkgUz6VLk,88
+quadra-2.6.1.dist-info/entry_points.txt,sha256=sRYonBZyx-sAJeWcQNQoVQIU5lm02cnCQt6b15k0WHU,43
+quadra-2.6.1.dist-info/licenses/LICENSE,sha256=8cTbQtcWa02YJoSpMeV_gxj3jpMTkxvl-w3WJ5gV_QE,11342
+quadra-2.6.1.dist-info/RECORD,,