code-loader 1.0.169.dev2__tar.gz → 1.0.169.dev4__tar.gz

{code_loader-1.0.169.dev2 → code_loader-1.0.169.dev4}/PKG-INFO

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

{code_loader-1.0.169.dev2 → code_loader-1.0.169.dev4}/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.dev2 → code_loader-1.0.169.dev4}/pyproject.toml

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