code-loader 1.0.87.dev1__py3-none-any.whl → 1.0.87.dev2__py3-none-any.whl

code_loader/contract/mapping.py

@@ -39,4 +39,14 @@ class NodeMapping:
 class NodeConnection:
     node: NodeMapping
     node_inputs: Optional[Dict[str, NodeMapping]]
-    prediction_type_name: Optional[str] = None
+    prediction_type_name: Optional[str] = None
+
+
+def leap_output(idx):
+    def dummy():
+        return None
+
+    node_mapping_type = NodeMappingType(f'Prediction{str(idx)}')
+    dummy.node_mapping = NodeMapping('', node_mapping_type)
+
+    return dummy

code_loader/inner_leap_binder/leapbinder_decorators.py

@@ -16,6 +16,16 @@ from code_loader.contract.visualizer_classes import LeapImage, LeapImageMask, Le
     LeapHorizontalBar, LeapImageWithBBox, LeapImageWithHeatmap
 
 
+def _add_mapping_connections(connects_to, arg_names, node_mapping_type, name):
+    for user_unique_name, connection_destinations in connects_to.items():
+        main_node_mapping = NodeMapping(name, node_mapping_type, user_unique_name, arg_names=arg_names)
+        node_inputs = {}
+        for arg_name, destination in zip(arg_names, connection_destinations):
+            node_inputs[arg_name] = destination.node_mapping
+
+        leap_binder.mapping_connections.append(NodeConnection(main_node_mapping, node_inputs))
+
+
 def tensorleap_custom_metric(name: str,
                              direction: Union[MetricDirection, Dict[str, MetricDirection]] = MetricDirection.Downward,
                              compute_insights: Optional[Union[bool, Dict[str, bool]]] = None,
@@ -32,15 +42,7 @@ def tensorleap_custom_metric(name: str,
 
         if connects_to is not None:
             arg_names = leap_binder.setup_container.metrics[-1].metric_handler_data.arg_names
-            for user_unique_name, connection_destinations in connects_to.items():
-                main_node_mapping = NodeMapping(name, NodeMappingType.Metric, user_unique_name, arg_names=arg_names)
-                node_inputs = {}
-                for arg_name, destination in zip(arg_names, connection_destinations):
-                    node_inputs[arg_name] = destination.node_mapping
-
-                leap_binder.mapping_connections.append(NodeConnection(main_node_mapping, node_inputs))
-
-
+            _add_mapping_connections(connects_to, arg_names, NodeMappingType.Metric, name)
 
         def _validate_input_args(*args, **kwargs) -> None:
             for i, arg in enumerate(args):
@@ -130,7 +132,8 @@ def tensorleap_custom_metric(name: str,
 
 
 def tensorleap_custom_visualizer(name: str, visualizer_type: LeapDataType,
-                                 heatmap_function: Optional[Callable[..., npt.NDArray[np.float32]]] = None):
+                                 heatmap_function: Optional[Callable[..., npt.NDArray[np.float32]]] = None,
+                                 connects_to=None):
     def decorating_function(user_function: VisualizerCallableInterface):
         for viz_handler in leap_binder.setup_container.visualizers:
             if viz_handler.visualizer_handler_data.name == name:
@@ -139,6 +142,10 @@ def tensorleap_custom_visualizer(name: str, visualizer_type: LeapDataType,
 
         leap_binder.set_visualizer(user_function, name, visualizer_type, heatmap_function)
 
+        if connects_to is not None:
+            arg_names = leap_binder.setup_container.visualizers[-1].visualizer_handler_data.arg_names
+            _add_mapping_connections(connects_to, arg_names, NodeMappingType.Visualizer, name)
+
         def _validate_input_args(*args, **kwargs):
             for i, arg in enumerate(args):
                 assert isinstance(arg, (np.ndarray, SamplePreprocessResponse)), (
@@ -328,6 +335,8 @@ def tensorleap_input_encoder(name: str, channel_dim=-1):
             _validate_result(result)
             return result
 
+        inner.node_mapping = NodeMapping(name, NodeMappingType.Input)
+
         return inner
 
     return decorating_function
@@ -377,7 +386,7 @@ def tensorleap_gt_encoder(name: str):
     return decorating_function
 
 
-def tensorleap_custom_loss(name: str):
+def tensorleap_custom_loss(name: str, connects_to=None):
     def decorating_function(user_function: CustomCallableInterface):
         for loss_handler in leap_binder.setup_container.custom_loss_handlers:
             if loss_handler.custom_loss_handler_data.name == name:
@@ -386,6 +395,11 @@ def tensorleap_custom_loss(name: str):
 
         leap_binder.add_custom_loss(user_function, name)
 
+        if connects_to is not None:
+            arg_names = leap_binder.setup_container.custom_loss_handlers[-1].custom_loss_handler_data.arg_names
+            _add_mapping_connections(connects_to, arg_names, NodeMappingType.CustomLoss, name)
+
+
         valid_types = (np.ndarray, SamplePreprocessResponse)
         try:
             import tensorflow as tf

{code_loader-1.0.87.dev1.dist-info → code_loader-1.0.87.dev2.dist-info}/METADATA

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

{code_loader-1.0.87.dev1.dist-info → code_loader-1.0.87.dev2.dist-info}/RECORD

@@ -4,7 +4,7 @@ code_loader/contract/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hS
 code_loader/contract/datasetclasses.py,sha256=ZN6XC-tg0qaT-GF4YiV2SsM2dUQW4fc1akLF-J4YT0w,7964
 code_loader/contract/enums.py,sha256=GEFkvUMXnCNt-GOoz7NJ9ecQZ2PPDettJNOsxsiM0wk,1622
 code_loader/contract/exceptions.py,sha256=jWqu5i7t-0IG0jGRsKF4DjJdrsdpJjIYpUkN1F4RiyQ,51
-code_loader/contract/mapping.py,sha256=j9xqgnDrQJZtNG00a3JnpoSEm2ezMCgE4TQdZC-NwkE,939
+code_loader/contract/mapping.py,sha256=qYTTw-UZUVGnyy2Coei0zxhsqM1XwWEn4JkLy_WswBU,1144
 code_loader/contract/responsedataclasses.py,sha256=RSx9m_R3LawhK5o1nAcO3hfp2F9oJYtxZr_bpP3bTmw,4005
 code_loader/contract/visualizer_classes.py,sha256=_nlukRfW8QeQaQG7G5HfAUSeCuoqslB4xJS2AGI_Nz8,15156
 code_loader/default_losses.py,sha256=NoOQym1106bDN5dcIk56Elr7ZG5quUHArqfP5-Nyxyo,1139
@@ -21,13 +21,13 @@ code_loader/experiment_api/utils.py,sha256=XZHtxge12TS4H4-8PjV3sKuhp8Ud6ojAiIzTZ
 code_loader/experiment_api/workingspace_config_utils.py,sha256=DLzXQCg4dgTV_YgaSbeTVzq-2ja_SQw4zi7LXwKL9cY,990
 code_loader/inner_leap_binder/__init__.py,sha256=koOlJyMNYzGbEsoIbXathSmQ-L38N_pEXH_HvL7beXU,99
 code_loader/inner_leap_binder/leapbinder.py,sha256=4mC6wvSKwfarDQbdSrISBG3uvDUIus_laXh1LlyCgK8,30932
-code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=6HRHVgLZuYBBssF_nreehaqKeB6V8gwvENBXHexpj4I,23434
+code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=sesjIgQ6fRFtNNEnKvxZHjX0l1uzkPqzvqB-DEomy5w,24170
 code_loader/leaploader.py,sha256=QPBG7xrNdzhnOiRtwUlVosyYcFNC_ivWgoS8nWOZ_5M,25768
 code_loader/leaploaderbase.py,sha256=VH0vddRmkqLtcDlYPCO7hfz1_VbKo43lUdHDAbd4iJc,4198
 code_loader/utils.py,sha256=aw2i_fqW_ADjLB66FWZd9DfpCQ7mPdMyauROC5Nd51I,2197
 code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/visualizers/default_visualizers.py,sha256=f95WzLYO2KU8endWxrmtwjHx7diDfm8AaWjucYHT7GI,2439
-code_loader-1.0.87.dev1.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.87.dev1.dist-info/METADATA,sha256=w9bosTgQZfkgg7SUfGLk8thr6vZM9glFr-LCYqOkvJc,854
-code_loader-1.0.87.dev1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-code_loader-1.0.87.dev1.dist-info/RECORD,,
+code_loader-1.0.87.dev2.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-1.0.87.dev2.dist-info/METADATA,sha256=WybWXrCev_1iopPHq08ykW4iCW2Uzrr881ELfqxDrJE,854
+code_loader-1.0.87.dev2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+code_loader-1.0.87.dev2.dist-info/RECORD,,