code-loader 1.0.87.dev5__tar.gz → 1.0.87.dev8__tar.gz

{code_loader-1.0.87.dev5 → code_loader-1.0.87.dev8}/PKG-INFO

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

{code_loader-1.0.87.dev5 → code_loader-1.0.87.dev8}/code_loader/contract/mapping.py

@@ -24,6 +24,7 @@ class NodeMappingType(Enum):
     Input3 = 'Input3'
     Input4 = 'Input4'
     Input5 = 'Input5'
+    PredictionLabels = 'PredictionLabels'
 
 
 @dataclass
@@ -39,7 +40,6 @@ class NodeMapping:
 class NodeConnection:
     node: NodeMapping
     node_inputs: Optional[Dict[str, NodeMapping]]
-    prediction_type_name: Optional[str] = None
 
 
 def leap_output(idx):
@@ -49,4 +49,5 @@ def leap_output(idx):
     node_mapping_type = NodeMappingType(f'Prediction{str(idx)}')
     dummy.node_mapping = NodeMapping('', node_mapping_type)
 
-    return dummy
+    return dummy
+

{code_loader-1.0.87.dev5 → code_loader-1.0.87.dev8}/code_loader/inner_leap_binder/leapbinder.py

@@ -12,7 +12,7 @@ from code_loader.contract.datasetclasses import SectionCallableInterface, InputH
     CustomMultipleReturnCallableInterfaceMultiArgs, DatasetBaseHandler, custom_latent_space_attribute, \
     RawInputsForHeatmap, VisualizerHandlerData, MetricHandlerData, CustomLossHandlerData, SamplePreprocessResponse
 from code_loader.contract.enums import LeapDataType, DataStateEnum, DataStateType, MetricDirection, DatasetMetadataType
-from code_loader.contract.mapping import NodeConnection
+from code_loader.contract.mapping import NodeConnection, NodeMapping, NodeMappingType
 from code_loader.contract.responsedataclasses import DatasetTestResultPayload
 from code_loader.contract.visualizer_classes import map_leap_data_type_to_visualizer_class
 from code_loader.default_losses import loss_name_to_function
@@ -310,7 +310,7 @@ class LeapBinder:
         metric_handler_data = MetricHandlerData(name, regular_arg_names, direction, compute_insights)
         self.setup_container.metrics.append(MetricHandler(metric_handler_data, function))
 
-    def add_prediction(self, name: str, labels: List[str], channel_dim: int = -1) -> None:
+    def add_prediction(self, name: str, labels: List[str], channel_dim: int = -1, prediction_index: Optional[int]=None) -> None:
         """
         Add prediction labels to the setup.
 
@@ -325,6 +325,11 @@ class LeapBinder:
             leap_binder.add_prediction(name='class_labels', labels=['cat', 'dog'])
         """
         self.setup_container.prediction_types.append(PredictionTypeHandler(name, labels, channel_dim))
+        if prediction_index is not None:
+            node_connection = NodeConnection(NodeMapping(name, NodeMappingType.PredictionLabels),
+                                             {'temp': NodeMapping('', NodeMappingType(f'Prediction{prediction_index}'))})
+
+            self.mapping_connections.append(node_connection)
 
     def set_ground_truth(self, function: SectionCallableInterface, name: str) -> None:
         """

{code_loader-1.0.87.dev5 → code_loader-1.0.87.dev8}/code_loader/inner_leap_binder/leapbinder_decorators.py

@@ -296,7 +296,7 @@ def tensorleap_unlabeled_preprocess():
     return decorating_function
 
 
-def tensorleap_input_encoder(name: str, channel_dim=-1):
+def tensorleap_input_encoder(name: str, channel_dim=-1, model_input_index=None):
     def decorating_function(user_function: SectionCallableInterface):
         for input_handler in leap_binder.setup_container.inputs:
             if input_handler.name == name:
@@ -335,7 +335,10 @@ def tensorleap_input_encoder(name: str, channel_dim=-1):
             _validate_result(result)
             return result
 
-        inner.node_mapping = NodeMapping(name, NodeMappingType.Input)
+        node_mapping_type = NodeMappingType.Input
+        if model_input_index is not None:
+            node_mapping_type = NodeMappingType(f'Input{str(model_input_index)}')
+        inner.node_mapping = NodeMapping(name, node_mapping_type)
 
         return inner
 

{code_loader-1.0.87.dev5 → code_loader-1.0.87.dev8}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "code-loader"
-version = "1.0.87.dev5"
+version = "1.0.87.dev8"
 description = ""
 authors = ["dorhar <doron.harnoy@tensorleap.ai>"]
 license = "MIT"