code-loader 1.0.174.dev3__tar.gz → 1.0.174.dev4__tar.gz

{code_loader-1.0.174.dev3 → code_loader-1.0.174.dev4}/PKG-INFO

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

{code_loader-1.0.174.dev3 → code_loader-1.0.174.dev4}/code_loader/inner_leap_binder/leapbinder_decorators.py

@@ -879,7 +879,8 @@ def tensorleap_custom_metric(name: str,
 
 def tensorleap_custom_instances_metric(name: str,
                              direction: Union[MetricDirection, Dict[str, MetricDirection]] = _UNSET,
-                             compute_insights: Optional[Union[bool, Dict[str, bool]]] = None):
+                             compute_insights: Optional[Union[bool, Dict[str, bool]]] = None,
+                             connects_to=None):
     name_to_unique_name = defaultdict(set)
 
     def decorating_function(
@@ -942,21 +943,21 @@ def tensorleap_custom_instances_metric(name: str,
                                 err_message +
                                 f"Invalid type for compute_insights['{k}']: expected bool, got type {type(v).__name__}."
                             )
-            # if connects_to is not None:
-            #     valid_types = (str, list, tuple, set)
-            #     if not isinstance(connects_to, valid_types):
-            #         raise TypeError(
-            #             err_message +
-            #             f"`connects_to` must be one of {valid_types}, got type {type(connects_to).__name__}."
-            #         )
-            #     if isinstance(connects_to, (list, tuple, set)):
-            #         invalid_elems = [f"{type(e).__name__}" for e in connects_to if not isinstance(e, str)]
-            #         if invalid_elems:
-            #             raise TypeError(
-            #                 err_message +
-            #                 f"All elements in `connects_to` must be strings, "
-            #                 f"but found element types: {invalid_elems}."
-            #             )
+            if connects_to is not None:
+                valid_types = (str, list, tuple, set)
+                if not isinstance(connects_to, valid_types):
+                    raise TypeError(
+                        err_message +
+                        f"`connects_to` must be one of {valid_types}, got type {type(connects_to).__name__}."
+                    )
+                if isinstance(connects_to, (list, tuple, set)):
+                    invalid_elems = [f"{type(e).__name__}" for e in connects_to if not isinstance(e, str)]
+                    if invalid_elems:
+                        raise TypeError(
+                            err_message +
+                            f"All elements in `connects_to` must be strings, "
+                            f"but found element types: {invalid_elems}."
+                        )
 
         _validate_decorators_signature()
 
@@ -1038,9 +1039,9 @@ def tensorleap_custom_instances_metric(name: str,
 
         leap_binder.add_custom_instance_metric(inner_without_validate, name, direction, compute_insights)
 
-        # if connects_to is not None:
-        #     arg_names = leap_binder.setup_container.metrics[-1].metric_handler_data.arg_names
-        #     _add_mapping_connections(connects_to, arg_names, NodeMappingType.Metric, name)
+        if connects_to is not None:
+            arg_names = leap_binder.setup_container.metrics[-1].metric_handler_data.arg_names
+            _add_mapping_connections(connects_to, arg_names, NodeMappingType.Metric, name)
 
         def inner(*args, **kwargs):
             if not _call_from_tl_platform:

{code_loader-1.0.174.dev3 → code_loader-1.0.174.dev4}/pyproject.toml

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