code-loader 1.0.208.dev6__tar.gz → 1.0.208.dev7__tar.gz

{code_loader-1.0.208.dev6 → code_loader-1.0.208.dev7}/PKG-INFO

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

{code_loader-1.0.208.dev6 → code_loader-1.0.208.dev7}/code_loader/contract/responsedataclasses.py

@@ -144,7 +144,6 @@ class LeapAnalysisConfiguration:
 class EngineFileContract:
     node_connections: Optional[List[NodeConnection]] = None
     leap_analysis_configuration: Optional[LeapAnalysisConfiguration] = None
-    latent_space_connections: Optional[List[NodeConnection]] = None
 
 
 @dataclass

{code_loader-1.0.208.dev6 → code_loader-1.0.208.dev7}/code_loader/inner_leap_binder/leapbinder.py

@@ -115,7 +115,6 @@ class LeapBinder:
         self._extend_with_default_losses()
 
         self.mapping_connections: List[NodeConnection] = []
-        self.latent_space_connections: List[NodeConnection] = []
         self.integration_test_func: Optional[Callable[[str, PreprocessResponse], Any]] = None
 
         self.batch_size_to_validate: Optional[int] = None

{code_loader-1.0.208.dev6 → code_loader-1.0.208.dev7}/code_loader/inner_leap_binder/leapbinder_decorators.py

@@ -2055,14 +2055,15 @@ def _decorate_model_latent_space(user_function, ls_name, use_ls_for_analysis, re
         ordered_connections = [kwargs[arg_name] for arg_name in arg_names if arg_name in kwargs]
         ordered_connections = list(args) + ordered_connections
 
-        leap_binder.latent_space_connections[:] = [
-            connection for connection in leap_binder.latent_space_connections
-            if connection.node.name != ls_name]
+        leap_binder.mapping_connections[:] = [
+            connection for connection in leap_binder.mapping_connections
+            if not (connection.node.type is NodeMappingType.CustomLatentSpace
+                    and connection.node.name == ls_name)]
         _add_mapping_connection(ls_name, ordered_connections, arg_names, ls_name,
-                                NodeMappingType.CustomLatentSpace,
-                                target_list=leap_binder.latent_space_connections)
+                                NodeMappingType.CustomLatentSpace)
         return None
 
+    @functools.wraps(user_function)
     def final_inner(*args, **kwargs):
         if os.environ.get(mapping_runtime_mode_env_var_mame):
             return mapping_inner(*args, **kwargs)

{code_loader-1.0.208.dev6 → code_loader-1.0.208.dev7}/code_loader/leaploader.py

@@ -396,8 +396,7 @@ class LeapLoader(LeapLoaderBase):
                                        print_log=print_log,
                                        engine_file_contract=EngineFileContract(
                                            global_leap_binder.mapping_connections,
-                                           global_leap_binder.leap_analysis_configuration,
-                                           global_leap_binder.latent_space_connections))
+                                           global_leap_binder.leap_analysis_configuration))
 
     def _check_integration_test_exists(self) -> DatasetTestResultPayload:
         test_result = DatasetTestResultPayload('integration_test')
@@ -457,9 +456,10 @@ class LeapLoader(LeapLoaderBase):
             )
         if global_leap_binder.setup_container.unlabeled_data_preprocess is not None:
             gt_bound_names = [
-                connection.node.name for connection in global_leap_binder.latent_space_connections
-                if any(node_input.type == NodeMappingType.GroundTruth
-                       for node_input in (connection.node_inputs or {}).values())
+                connection.node.name for connection in global_leap_binder.mapping_connections
+                if connection.node.type is NodeMappingType.CustomLatentSpace
+                and any(node_input.type == NodeMappingType.GroundTruth
+                        for node_input in (connection.node_inputs or {}).values())
             ]
             if gt_bound_names:
                 test_result.display[TestingSectionEnum.Warnings.name] = (

{code_loader-1.0.208.dev6 → code_loader-1.0.208.dev7}/pyproject.toml

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