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

code_loader/inner_leap_binder/leapbinder_decorators.py

@@ -773,12 +773,6 @@ def tensorleap_custom_metric(name: str,
         except (TypeError, ValueError):
             pass
 
-        leap_binder.add_custom_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)
-
         def inner(*args, **kwargs):
             if not _call_from_tl_platform:
                 set_current('tensorleap_custom_metric')
@@ -808,6 +802,8 @@ def tensorleap_custom_metric(name: str,
 
             return None
 
+        leap_binder.add_custom_metric(inner, name, direction, compute_insights)
+
         mapping_inner.arg_names = leap_binder.setup_container.metrics[-1].metric_handler_data.arg_names
         mapping_inner.name = name
 
@@ -817,6 +813,10 @@ def tensorleap_custom_metric(name: str,
             else:
                 return inner(*args, **kwargs)
 
+        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)
+
         return final_inner
 
     return decorating_function
@@ -1813,5 +1813,3 @@ def tensorleap_status_table():
 if not _call_from_tl_platform:
     set_current, update_env_params_func = tensorleap_status_table()
 
-
-

{code_loader-1.0.169.dev3.dist-info → code_loader-1.0.169.dev4.dist-info}/METADATA

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

{code_loader-1.0.169.dev3.dist-info → code_loader-1.0.169.dev4.dist-info}/RECORD

@@ -22,7 +22,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=Wu8Mgltf46YkqNczmYhpY-7aYIbQwTJ87hEDoyo_Mzo,33333
-code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=No2uivIsQOwam0gx0R-m3pXQj6trFlVXyPQvX8-hzhI,85543
+code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=KeR6ubKBp9tOdAKcvnUkB0yDiYxvus8CZQqHmR3sj8w,85524
 code_loader/leaploader.py,sha256=TW82exQGqEk36XOLonjOfx4zU4P0BoOQLbft687uU_0,30707
 code_loader/leaploaderbase.py,sha256=NXCDIIF7-ziGJccKIE9NszMSYKEE-3bn4Z4Xa3oYYOc,5909
 code_loader/mixpanel_tracker.py,sha256=eKvymkw7X2Ht6iw-a0V9VQm6OnB9kW7hYy35YtwRAvU,8457
@@ -32,7 +32,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.169.dev3.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.169.dev3.dist-info/METADATA,sha256=r30xqP7qNHBKH_n3g4FAKYYkuwkpzaWSM5fHKUDubvc,1095
-code_loader-1.0.169.dev3.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-code_loader-1.0.169.dev3.dist-info/RECORD,,
+code_loader-1.0.169.dev4.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-1.0.169.dev4.dist-info/METADATA,sha256=W2xOkBo2kpPTVr2eQNbS55Skc0XxwKB-VHuoWxDbHSg,1095
+code_loader-1.0.169.dev4.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+code_loader-1.0.169.dev4.dist-info/RECORD,,