code-loader 1.0.170__tar.gz → 1.0.172.dev1__tar.gz

{code_loader-1.0.170 → code_loader-1.0.172.dev1}/PKG-INFO

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

{code_loader-1.0.170 → code_loader-1.0.172.dev1}/code_loader/contract/datasetclasses.py

@@ -1,6 +1,6 @@
 import warnings
 from dataclasses import dataclass, field
-from typing import Any, Callable, List, Optional, Dict, Union, Type, Literal
+from typing import Any, Callable, Iterable, List, Optional, Dict, Union, Type, Literal, Tuple
 import re
 import numpy as np
 import numpy.typing as npt
@@ -165,6 +165,36 @@ class MetricHandlerData:
     compute_insights: Optional[Union[bool, Dict[str, bool]]] = None
 
 
+def normalize_metric_handler_settings(
+        direction: Union[None, MetricDirection, Dict[str, MetricDirection]],
+        compute_insights: Optional[Union[bool, Dict[str, bool]]],
+        result_keys: Iterable[str],
+) -> Tuple[Union[None, MetricDirection, Dict[str, MetricDirection]], Optional[Union[bool, Dict[str, bool]]]]:
+    ordered_result_keys = list(result_keys)
+    if not ordered_result_keys:
+        return direction, compute_insights
+
+    normalized_direction = direction
+    if isinstance(direction, MetricDirection):
+        normalized_direction = {key: direction for key in ordered_result_keys}
+    elif isinstance(direction, dict):
+        normalized_direction = {
+            key: direction.get(key, MetricDirection.Downward)
+            for key in ordered_result_keys
+        }
+
+    normalized_compute_insights = compute_insights
+    if isinstance(compute_insights, bool):
+        normalized_compute_insights = {key: compute_insights for key in ordered_result_keys}
+    elif isinstance(compute_insights, dict):
+        normalized_compute_insights = {
+            key: compute_insights.get(key, True)
+            for key in ordered_result_keys
+        }
+
+    return normalized_direction, normalized_compute_insights
+
+
 @dataclass
 class MetricHandler:
     metric_handler_data: MetricHandlerData
@@ -272,4 +302,3 @@ class DatasetSample:
     state: DataStateEnum
     custom_latent_space: Optional[npt.NDArray[np.float32]] = None
     instance_masks: Optional[Dict[str, ElementInstance]] = None
-

{code_loader-1.0.170 → code_loader-1.0.172.dev1}/code_loader/inner_leap_binder/leapbinder_decorators.py

@@ -961,9 +961,8 @@ def tensorleap_metadata(
         def _validate_result(result):
             supported_result_types = (type(None), int, str, bool, float, dict, np.floating,
                                       np.bool_, np.unsignedinteger, np.signedinteger, np.integer)
-            if isinstance(result, tuple):
-                validate_output_structure(result, func_name=user_function.__name__,
-                                          expected_type_name=supported_result_types)
+            validate_output_structure(result, func_name=user_function.__name__,
+                                      expected_type_name=supported_result_types)
             assert isinstance(result, supported_result_types), \
                 (f'{user_function.__name__}() validation failed: '
                  f'Unsupported return type. Got {type(result)}. should be any of {str(supported_result_types)}')

{code_loader-1.0.170 → code_loader-1.0.172.dev1}/code_loader/mixpanel_tracker.py

@@ -13,6 +13,7 @@ import mixpanel  # type: ignore[import]
 logger = logging.getLogger(__name__)
 
 TRACKING_VERSION = '1'
+TL_DISABLE_ANALYTICS_ENV_VAR = 'TL_DISABLE_ANALYTICS'
 
 
 class AnalyticsEvent(str, Enum):
@@ -143,6 +144,10 @@ class MixpanelTracker:
         # Skip tracking if IS_TENSORLEAP_PLATFORM environment variable is set to 'true'
         if os.environ.get('IS_TENSORLEAP_PLATFORM') == 'true':
             return
+
+        # Skip tracking if TL_DISABLE_ANALYTICS environment variable is set
+        if os.environ.get(TL_DISABLE_ANALYTICS_ENV_VAR):
+            return
             
         try:
             distinct_id = self._get_distinct_id()
@@ -199,11 +204,24 @@ def get_tracker() -> MixpanelTracker:
     return _tracker
 
 
+def _is_analytics_disabled() -> bool:
+    return bool(os.environ.get(TL_DISABLE_ANALYTICS_ENV_VAR))
+
+
+def disable_analytics() -> None:
+    """Programmatically disable all analytics/Mixpanel tracking."""
+    os.environ[TL_DISABLE_ANALYTICS_ENV_VAR] = 'true'
+
+
 def track_code_loader_loaded(event_properties: Optional[Dict[str, Any]] = None) -> None:
+    if _is_analytics_disabled():
+        return
     get_tracker().track_code_loader_loaded(event_properties)
 
 
 def track_integration_test_event(event_name: Union[str, AnalyticsEvent], event_properties: Optional[Dict[str, Any]] = None) -> None:
+    if _is_analytics_disabled():
+        return
     get_tracker().track_integration_test_event(event_name, event_properties)
 
 
@@ -213,6 +231,8 @@ _integration_events_emitted: Set[str] = set()
 
 def emit_integration_event_once(event_name: Union[str, AnalyticsEvent], props: Dict[str, Any]) -> None:
     """Emit an integration test event only once per test run."""
+    if _is_analytics_disabled():
+        return
     event_name_str = str(event_name)
     if event_name_str in _integration_events_emitted:
         return

{code_loader-1.0.170 → code_loader-1.0.172.dev1}/pyproject.toml

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