congrads 1.1.0__py3-none-any.whl → 1.1.2__py3-none-any.whl

congrads/descriptor.py

@@ -122,7 +122,7 @@ class Descriptor:
         """
         return tag in self._tag_to_key and tag in self._tag_to_index
 
-    def location(self, tag: str) -> tuple[str, int]:
+    def location(self, tag: str) -> tuple[str, int | None]:
         """Get the key and index for a given tag.
 
         Looks up the mapping for a registered tag and returns the associated
@@ -132,16 +132,16 @@ class Descriptor:
             tag (str): The tag identifier. Must be registered.
 
         Returns:
-            tuple ((str, int)): A tuple containing:
+            tuple ((str, int | None)): A tuple containing:
                 - The key in the data dictionary which holds the data (str).
-                - The tensor index where the data is present (int).
+                - The tensor index where the data is present or None (int | None).
 
         Raises:
             ValueError: If the tag is not registered in the descriptor.
         """
         key = self._tag_to_key.get(tag)
         index = self._tag_to_index.get(tag)
-        if key is None or index is None:
+        if key is None:
             raise ValueError(f"Tag '{tag}' is not registered in descriptor.")
         return key, index
 
@@ -150,6 +150,7 @@ class Descriptor:
 
         Retrieves the key and index associated with a tag and selects
         the corresponding slice from the given prediction tensor.
+        Returns the full tensor if no index was specified when registering the tag.
 
         Args:
             tag (str): The tag identifier. Must be registered.
@@ -157,10 +158,12 @@ class Descriptor:
 
         Returns:
             Tensor: A tensor slice of shape ``(batch_size, 1)`` containing
-            the predictions for the specified tag.
+            the predictions for the specified tag, or the full tensor if no index was specified when registering the tag.
 
         Raises:
             ValueError: If the tag is not registered in the descriptor.
         """
         key, index = self.location(tag)
+        if index is None:
+            return data[key]
         return data[key][:, index : index + 1]

{congrads-1.1.0.dist-info → congrads-1.1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: congrads
-Version: 1.1.0
+Version: 1.1.2
 Summary: A toolbox for using Constraint Guided Gradient Descent when training neural networks.
 Author: Wout Rombouts, Quinten Van Baelen, Peter Karsmakers
 Author-email: Wout Rombouts <wout.rombouts@kuleuven.be>, Quinten Van Baelen <quinten.vanbaelen@kuleuven.be>, Peter Karsmakers <peter.karsmakers@kuleuven.be>
@@ -36,7 +36,7 @@ Requires-Dist: torch>=2.0.0
 Requires-Dist: torchvision>=0.15.1
 Requires-Dist: tqdm>=4.65.0
 Requires-Dist: matplotlib>=3.7.0 ; extra == 'examples'
-Requires-Dist: tensorboard>=2.12.0 ; extra == 'examples'
+Requires-Dist: tensorboard>=2.18.0 ; extra == 'examples'
 Requires-Python: >=3.11
 Provides-Extra: examples
 Description-Content-Type: text/markdown

{congrads-1.1.0.dist-info → congrads-1.1.2.dist-info}/RECORD

@@ -3,12 +3,12 @@ congrads/checkpoints.py,sha256=VKSl8gOtp_VPF4SVMDdj34WuAJqsv5zVtf2YlSDVOqE,7217
 congrads/constraints.py,sha256=vtYt1rpmYCekWWES_85L8hgQAkJROP-R5asuizpBQqE,52336
 congrads/core.py,sha256=_Ah2EE9WxJL7laUBcoMcgGWuZj3Hnuby4Mu4HOfkT9c,36161
 congrads/datasets.py,sha256=RfffRiA7Qijc69cJTBJhItTZ8x9B-p1kXMjvcfEC_nA,31102
-congrads/descriptor.py,sha256=Z_qGtNMQ3BKYjQpUNJxlsmEWBDwYk98ozig7nwm7RkI,6681
+congrads/descriptor.py,sha256=MXaLPt3jOLbHR_p7ZHrBMPYfrxyxyYqU5hvte-Sqfuk,6903
 congrads/metrics.py,sha256=QxQiLIVaCV1yyryXsKOb1HKtfYFnHNIJMG8fUDG_1JE,4614
 congrads/networks.py,sha256=UPzPDU0wI2zoOEvi697QBSDOtaa3Rc0rgCb-tCxbjak,2252
 congrads/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 congrads/transformations.py,sha256=57xXmf87eUHmmLUqzt0bQZdDlAc0SFxShXK6Bku1z34,3327
 congrads/utils.py,sha256=Bqf5gdioBNot8IhQR298n6WHpQZO8W_Ex_VwB96tlmc,41508
-congrads-1.1.0.dist-info/WHEEL,sha256=Pi5uDq5Fdo_Rr-HD5h9BiPn9Et29Y9Sh8NhcJNnFU1c,79
-congrads-1.1.0.dist-info/METADATA,sha256=336e2CyqdhojvWovJaHrBuN1OekXA5GqU28_91GMvVY,10534
-congrads-1.1.0.dist-info/RECORD,,
+congrads-1.1.2.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
+congrads-1.1.2.dist-info/METADATA,sha256=6Hbb_tlG8-4CMXAgkHN3SO_vcx-ntwZc94GB9tMJCEQ,10534
+congrads-1.1.2.dist-info/RECORD,,

{congrads-1.1.0.dist-info → congrads-1.1.2.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: uv 0.8.17
+Generator: uv 0.8.24
 Root-Is-Purelib: true
 Tag: py3-none-any