code-loader 1.0.170__tar.gz → 1.0.171__tar.gz

{code_loader-1.0.170 → code_loader-1.0.171}/PKG-INFO

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

{code_loader-1.0.170 → code_loader-1.0.171}/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.171}/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.171}/pyproject.toml

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