code-loader 1.0.186__tar.gz → 1.0.186.dev0__tar.gz

{code_loader-1.0.186 → code_loader-1.0.186.dev0}/PKG-INFO

@@ -1,7 +1,8 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.1
 Name: code-loader
-Version: 1.0.186
+Version: 1.0.186.dev0
 Summary: 
+Home-page: https://github.com/tensorleap/code-loader
 License: MIT
 Author: dorhar
 Author-email: doron.harnoy@tensorleap.ai
@@ -19,7 +20,6 @@ Requires-Dist: numpy (>=2.3.2,<3.0.0) ; python_version >= "3.11" and python_vers
 Requires-Dist: psutil (>=5.9.5,<6.0.0)
 Requires-Dist: pyyaml (>=6.0.2,<7.0.0)
 Requires-Dist: requests (>=2.32.3,<3.0.0)
-Project-URL: Homepage, https://github.com/tensorleap/code-loader
 Project-URL: Repository, https://github.com/tensorleap/code-loader
 Description-Content-Type: text/markdown
 

{code_loader-1.0.186 → code_loader-1.0.186.dev0}/code_loader/inner_leap_binder/leapbinder_decorators.py

@@ -768,14 +768,20 @@ def tensorleap_custom_metric(name: str,
 
                 leap_binder.setup_container.metrics[-1].metric_handler_data.direction = effective_direction
 
+                # Broadcast a single `compute_insights` value across all dict keys (and fill missing keys
+                # for a partial dict), so it applies to every metric in the returned dict.
+                if compute_insights is None:
+                    effective_compute_insights = None
+                elif isinstance(compute_insights, dict):
+                    effective_compute_insights = {key: compute_insights.get(key, True) for key in result_keys}
+                else:
+                    effective_compute_insights = {key: compute_insights for key in result_keys}
+
+                leap_binder.setup_container.metrics[-1].metric_handler_data.compute_insights = effective_compute_insights
+
                 if defaulted_direction_keys and not _call_from_tl_platform:
-                    if compute_insights is None:
-                        effective_compute_insights = {}
-                    elif isinstance(compute_insights, dict):
-                        effective_compute_insights = compute_insights
-                    else:
-                        effective_compute_insights = {k: compute_insights for k in result_keys}
-                    warning_keys = {key for key in defaulted_direction_keys if effective_compute_insights.get(key, True)}
+                    warning_keys = {key for key in defaulted_direction_keys
+                                    if (effective_compute_insights or {}).get(key, True)}
                     if warning_keys:
                         store_warning_by_param(
                             param_name=f"direction[{warning_keys}]",

{code_loader-1.0.186 → code_loader-1.0.186.dev0}/pyproject.toml

@@ -1,7 +1,7 @@
 [tool.poetry]
 name = "code-loader"
 
-version = "1.0.186"
+version = "1.0.186.dev0"
 description = ""
 authors = ["dorhar <doron.harnoy@tensorleap.ai>"]
 license = "MIT"