code-loader 1.0.174.dev3__py3-none-any.whl → 1.0.174.dev4__py3-none-any.whl

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.dist-info → code_loader-1.0.174.dev4.dist-info}/METADATA

@@ -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.dist-info → code_loader-1.0.174.dev4.dist-info}/RECORD

@@ -21,7 +21,7 @@ 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=31H-poFoC16QmBtwLIncLD7SeyiSIrr2MKNCo2YomWw,35944
-code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=6GiiP1uHnSE_0vjd0vhkPq24emGAfy5qnPeeKEMNwJ4,99574
+code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=lRgfWBZ7yHrybeaii0mNUnL37mFR3DmezPhUZaxXHHo,99585
 code_loader/leaploader.py,sha256=TW82exQGqEk36XOLonjOfx4zU4P0BoOQLbft687uU_0,30707
 code_loader/leaploaderbase.py,sha256=NXCDIIF7-ziGJccKIE9NszMSYKEE-3bn4Z4Xa3oYYOc,5909
 code_loader/mixpanel_tracker.py,sha256=U7eUGrPjc-2rgFG7isqosf65tKZkotQ0XKuAML_lIjA,9067
@@ -31,7 +31,7 @@ code_loader/plot_functions/visualize.py,sha256=gsBAYYkwMh7jIpJeDMPS8G4CW-pxwx6Lz
 code_loader/utils.py,sha256=YecipkdTA-VcE9F0RQcY9cFnY8P3AksPnHM2Db7xUSk,3972
 code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/visualizers/default_visualizers.py,sha256=onRnLE_TXfgLN4o52hQIOOhUcFexGlqJ3xSpQDVLuZM,2604
-code_loader-1.0.174.dev3.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.174.dev3.dist-info/METADATA,sha256=GGMVEl4xJINLriZeIjBCUK_HBNm_WCKBoVLggyX1Ilk,1095
-code_loader-1.0.174.dev3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-code_loader-1.0.174.dev3.dist-info/RECORD,,
+code_loader-1.0.174.dev4.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-1.0.174.dev4.dist-info/METADATA,sha256=cmh1wefhdQLdpgo2-ETdjJ_fQ4VCguAd2Wyn44xSXhg,1095
+code_loader-1.0.174.dev4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+code_loader-1.0.174.dev4.dist-info/RECORD,,