code-loader 1.0.35__py3-none-any.whl → 1.0.36.dev1__py3-none-any.whl

code_loader/contract/datasetclasses.py

@@ -89,6 +89,10 @@ class MetricHandler:
     arg_names: List[str]
     direction: Optional[MetricDirection] = MetricDirection.Downward
 
+@dataclass
+class RawInputsForHeatmap:
+    raw_input_by_vizualizer_arg_name: Dict[str, np.array]
+
 
 @dataclass
 class VisualizerHandler:

code_loader/inner_leap_binder/leapbinder.py

@@ -9,7 +9,7 @@ from code_loader.contract.datasetclasses import SectionCallableInterface, InputH
     PreprocessHandler, VisualizerCallableInterface, CustomLossHandler, CustomCallableInterface, PredictionTypeHandler, \
     MetadataSectionCallableInterface, UnlabeledDataPreprocessHandler, CustomLayerHandler, MetricHandler, \
     CustomCallableInterfaceMultiArgs, ConfusionMatrixCallableInterfaceMultiArgs, VisualizerCallableReturnType, \
-    CustomMultipleReturnCallableInterfaceMultiArgs, DatasetBaseHandler, custom_latent_space_attribute
+    CustomMultipleReturnCallableInterfaceMultiArgs, DatasetBaseHandler, custom_latent_space_attribute, RawInputsForHeatmap
 from code_loader.contract.enums import LeapDataType, DataStateEnum, DataStateType, MetricDirection
 from code_loader.contract.responsedataclasses import DatasetTestResultPayload
 from code_loader.contract.visualizer_classes import map_leap_data_type_to_visualizer_class
@@ -50,10 +50,16 @@ class LeapBinder:
                        heatmap_visualizer: Optional[Callable[..., npt.NDArray[np.float32]]] = None) -> None:
         arg_names = inspect.getfullargspec(function)[0]
         if heatmap_visualizer:
-            if arg_names != inspect.getfullargspec(heatmap_visualizer)[0]:
-                raise Exception(
-                    f'The argument names of the heatmap visualizer callback must match the visualizer callback '
-                    f'{str(arg_names)}')
+            visualizer_arg_names_set = set(arg_names)
+            heatmap_visualizer_inspection = inspect.getfullargspec(heatmap_visualizer)
+            heatmap_arg_names_set = set(heatmap_visualizer_inspection[0])
+            if visualizer_arg_names_set != heatmap_arg_names_set:
+                arg_names_difference = set(inspect.getfullargspec(heatmap_visualizer)[0]).difference(set(arg_names))
+                if len(arg_names_difference) != 1 or \
+                        heatmap_visualizer_inspection.annotations[list(arg_names_difference)[0]] != RawInputsForHeatmap:
+                    raise Exception(
+                        f'The argument names of the heatmap visualizer callback must match the visualizer callback '
+                        f'{str(arg_names)}')
 
         if visualizer_type.value not in map_leap_data_type_to_visualizer_class:
             raise Exception(

{code_loader-1.0.35.dist-info → code_loader-1.0.36.dev1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: code-loader
-Version: 1.0.35
+Version: 1.0.36.dev1
 Summary: 
 Home-page: https://github.com/tensorleap/code-loader
 License: MIT

{code_loader-1.0.35.dist-info → code_loader-1.0.36.dev1.dist-info}/RECORD

@@ -2,18 +2,18 @@ LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
 code_loader/__init__.py,sha256=V3DEXSN6Ie6PlGeSAbzjp9ufRj0XPJLpD7pDLLYxk6M,122
 code_loader/code_inegration_processes_manager.py,sha256=XslWOPeNQk4RAFJ_f3tP5Oe3EgcIR7BE7Y8r9Ty73-o,3261
 code_loader/contract/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-code_loader/contract/datasetclasses.py,sha256=dlLUblktFlNhQ6_6CaRiSxvrHU07fWTpnTXMPDbojUI,4599
+code_loader/contract/datasetclasses.py,sha256=Vey-Q9k4s6FzuGCWERtryL44jBYsNr8pCUxqUsjYrrg,4696
 code_loader/contract/enums.py,sha256=wJSMFXL_E-JWK_XAgsqOZwOLDiRfcT_rp0oQ3P_edNI,1560
 code_loader/contract/exceptions.py,sha256=jWqu5i7t-0IG0jGRsKF4DjJdrsdpJjIYpUkN1F4RiyQ,51
 code_loader/contract/responsedataclasses.py,sha256=WSHmFZWOFhGL1eED1u-aoRotPQg2owFQ-t3xSViWXSI,2808
 code_loader/contract/visualizer_classes.py,sha256=1FjVO744J_EMuJfHWXGdvSz6vl3Vu7iS3CDfs8MzEEQ,5138
 code_loader/inner_leap_binder/__init__.py,sha256=koOlJyMNYzGbEsoIbXathSmQ-L38N_pEXH_HvL7beXU,99
-code_loader/inner_leap_binder/leapbinder.py,sha256=JBf0iWz-3spv6ZKCvYa2aDzqatNoP6kYAfDPmNdjrHU,13078
+code_loader/inner_leap_binder/leapbinder.py,sha256=cayCJxnksr5lrEIX1HngiuPpTupStti4DD6KX3k-PBI,13610
 code_loader/leaploader.py,sha256=pUySweZetJ6SsubCcZlDCJpvWmUrm5YlPlkZWQxY1hQ,17289
 code_loader/utils.py,sha256=61I4PgSl-ZBIe4DifLxMNlBELE-HQR2pB9efVYPceIU,2230
 code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/visualizers/default_visualizers.py,sha256=HqWx2qfTrroGl2n8Fpmr_4X-rk7tE2oGapjO3gzz4WY,2226
-code_loader-1.0.35.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.35.dist-info/METADATA,sha256=77FVSzalYJVewUYCf4h1zHF0IjurHFTLgseSQ0snzxQ,768
-code_loader-1.0.35.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-code_loader-1.0.35.dist-info/RECORD,,
+code_loader-1.0.36.dev1.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-1.0.36.dev1.dist-info/METADATA,sha256=O9W3K9okn4L95N12M_y4x9xGQYZGet7FdQ1Ljql4p24,773
+code_loader-1.0.36.dev1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+code_loader-1.0.36.dev1.dist-info/RECORD,,

{code_loader-1.0.35.dist-info → code_loader-1.0.36.dev1.dist-info}/WHEEL

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