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

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

@@ -1,8 +1,7 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: code-loader
-Version: 1.0.187
+Version: 1.0.187.dev0
 Summary: 
-Home-page: https://github.com/tensorleap/code-loader
 License: MIT
 Author: dorhar
 Author-email: doron.harnoy@tensorleap.ai
@@ -20,6 +19,7 @@ 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.187 → code_loader-1.0.187.dev0}/code_loader/inner_leap_binder/leapbinder.py

@@ -31,6 +31,31 @@ from code_loader.visualizers.default_visualizers import DefaultVisualizer, \
 mapping_runtime_mode_env_var_mame = '__MAPPING_RUNTIME_MODE__'
 
 
+def _stringized_annotation_type_name(annotation: Any) -> Optional[str]:
+    """Return the bare type name when ``annotation`` is a *stringized* annotation
+    (e.g. produced by ``from __future__ import annotations`` or a quoted hint),
+    otherwise ``None``. code_loader inspects raw annotations expecting real classes,
+    so a stringized annotation silently breaks type detection."""
+    if isinstance(annotation, str):
+        return annotation.split("[")[0].strip().rsplit(".", 1)[-1]
+    return None
+
+
+def _reject_stringized_sample_preprocess_response(function: Callable, arg_name: str, annotation: Any) -> None:
+    """Fail fast when a ``SamplePreprocessResponse`` argument was stringized. Otherwise
+    ``arg_type == SamplePreprocessResponse`` silently evaluates to ``False`` and the
+    argument is mis-wired as a regular input — a bug that only surfaces on the platform.
+    Fires only on the actual damage (a stringized SamplePreprocessResponse), not merely
+    because the file uses future annotations."""
+    if _stringized_annotation_type_name(annotation) == SamplePreprocessResponse.__name__:
+        raise Exception(
+            f"Argument '{arg_name}' of function '{function.__name__}' is annotated with a string "
+            f"('{annotation}') instead of the SamplePreprocessResponse type. This usually means the "
+            f"file uses 'from __future__ import annotations' (or a quoted hint), which stringizes "
+            f"annotations and breaks Tensorleap type detection. Remove that import (or the quotes) so "
+            f"SamplePreprocessResponse is referenced as a real type.")
+
+
 
 
 class LeapBinder:
@@ -123,6 +148,7 @@ class LeapBinder:
         regular_arg_names = inspect.getfullargspec(function)[0]
         preprocess_response_arg_name = None
         for arg_name, arg_type in inspect.getfullargspec(function).annotations.items():
+            _reject_stringized_sample_preprocess_response(function, arg_name, arg_type)
             if arg_type == SamplePreprocessResponse:
                 if preprocess_response_arg_name is not None:
                     raise Exception("only one argument can be of type SamplePreprocessResponse")
@@ -155,6 +181,13 @@ class LeapBinder:
                   f"https://docs.python.org/3/library/typing.html")
         else:
             return_type = func_annotations["return"]
+            if isinstance(return_type, str):
+                raise Exception(
+                    f"The return type hint of function '{function.__name__}' is a string "
+                    f"('{return_type}') instead of a type. This usually means the file uses "
+                    f"'from __future__ import annotations' (or a quoted return hint), which stringizes "
+                    f"annotations and prevents Tensorleap from detecting the visualizer type. Remove "
+                    f"that import (or the quotes) so the return type is a real class.")
             if return_type not in LeapData.__args__:  # type: ignore[attr-defined]
                 raise Exception(
                     f'The return type of function {function.__name__} is invalid. current return type: {return_type}, '  # type: ignore[attr-defined]
@@ -296,6 +329,7 @@ class LeapBinder:
         regular_arg_names = inspect.getfullargspec(function)[0]
         preprocess_response_arg_name = None
         for arg_name, arg_type in inspect.getfullargspec(function).annotations.items():
+            _reject_stringized_sample_preprocess_response(function, arg_name, arg_type)
             if arg_type == SamplePreprocessResponse:
                 if preprocess_response_arg_name is not None:
                     raise Exception("only one argument can be of type SamplePreprocessResponse")
@@ -339,6 +373,7 @@ class LeapBinder:
         regular_arg_names = inspect.getfullargspec(function)[0]
         preprocess_response_arg_name = None
         for arg_name, arg_type in inspect.getfullargspec(function).annotations.items():
+            _reject_stringized_sample_preprocess_response(function, arg_name, arg_type)
             if arg_type == SamplePreprocessResponse:
                 if preprocess_response_arg_name is not None:
                     raise Exception("only one argument can be of type SamplePreprocessResponse")
@@ -380,6 +415,7 @@ class LeapBinder:
         regular_arg_names = inspect.getfullargspec(function)[0]
         preprocess_response_arg_name = None
         for arg_name, arg_type in inspect.getfullargspec(function).annotations.items():
+            _reject_stringized_sample_preprocess_response(function, arg_name, arg_type)
             if arg_type == SamplePreprocessResponse:
                 if preprocess_response_arg_name is not None:
                     raise Exception("only one argument can be of type SamplePreprocessResponse")

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

@@ -768,18 +768,14 @@ def tensorleap_custom_metric(name: str,
 
                 leap_binder.setup_container.metrics[-1].metric_handler_data.direction = effective_direction
 
-                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:
-                    warning_keys = {key for key in defaulted_direction_keys
-                                    if (effective_compute_insights or {}).get(key, True)}
+                    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)}
                     if warning_keys:
                         store_warning_by_param(
                             param_name=f"direction[{warning_keys}]",

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

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