code-loader 1.0.87.dev4__py3-none-any.whl → 1.0.87.dev7__py3-none-any.whl

code_loader/contract/mapping.py

@@ -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/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/leaploader.py

@@ -19,7 +19,8 @@ from code_loader.contract.enums import DataStateEnum, TestingSectionEnum, DataSt
 from code_loader.contract.exceptions import DatasetScriptException
 from code_loader.contract.responsedataclasses import DatasetIntegParseResult, DatasetTestResultPayload, \
     DatasetPreprocess, DatasetSetup, DatasetInputInstance, DatasetOutputInstance, DatasetMetadataInstance, \
-    VisualizerInstance, PredictionTypeInstance, ModelSetup, CustomLayerInstance, MetricInstance, CustomLossInstance
+    VisualizerInstance, PredictionTypeInstance, ModelSetup, CustomLayerInstance, MetricInstance, CustomLossInstance, \
+    EngineFileContract
 from code_loader.inner_leap_binder import global_leap_binder
 from code_loader.leaploaderbase import LeapLoaderBase
 from code_loader.utils import get_root_exception_file_and_line_number
@@ -180,7 +181,7 @@ class LeapLoader(LeapLoaderBase):
                                        is_valid_for_model=is_valid_for_model, setup=setup_response,
                                        model_setup=model_setup, general_error=general_error,
                                        print_log=print_log,
-                                       engine_file_contract=global_leap_binder.mapping_connections)
+                                       engine_file_contract=EngineFileContract(global_leap_binder.mapping_connections))
 
     def _check_preprocess(self) -> DatasetTestResultPayload:
         test_result = DatasetTestResultPayload('preprocess')

{code_loader-1.0.87.dev4.dist-info → code_loader-1.0.87.dev7.dist-info}/METADATA

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

{code_loader-1.0.87.dev4.dist-info → code_loader-1.0.87.dev7.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=qYTTw-UZUVGnyy2Coei0zxhsqM1XwWEn4JkLy_WswBU,1144
+code_loader/contract/mapping.py,sha256=H7DB_Km9yvu3d80O1EPJ47tFF4s4axollfBaeVBeZsA,1146
 code_loader/contract/responsedataclasses.py,sha256=6-5DJkYBdXb3UB1eNidTTPPBIYxMjEoMdYDkp9VhH8o,4223
 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=sesjIgQ6fRFtNNEnKvxZHjX0l1uzkPqzvqB-DEomy5w,24170
-code_loader/leaploader.py,sha256=Te3W1FayB3sVq0hwBLIX74P51lFYQ6XPIE8TSJ-RVmY,25868
+code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=y5TuhJe-J3xEE0Oj7bxIfhyN1lXiLJgfgy3ZiAs-yic,24364
+code_loader/leaploader.py,sha256=3_opkzRnzGWqB8aubSqpCEf1aPo3Z7nkKY4R5j8iFeY,25914
 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.dev4.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.87.dev4.dist-info/METADATA,sha256=n8R93u2M5hbt473H1vvgVpwzH0Vfg1Lyjktbv6QF_Ck,854
-code_loader-1.0.87.dev4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-code_loader-1.0.87.dev4.dist-info/RECORD,,
+code_loader-1.0.87.dev7.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-1.0.87.dev7.dist-info/METADATA,sha256=T4t4-pJgfmfy3Z77joe6EIqm9BHVb2JfQ9G5yv0NPX0,854
+code_loader-1.0.87.dev7.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+code_loader-1.0.87.dev7.dist-info/RECORD,,