code-loader 1.0.114__py3-none-any.whl → 1.0.116__py3-none-any.whl

code_loader/__init__.py

@@ -5,6 +5,6 @@ from code_loader.experiment_api.client import Client
 
 try:
     from code_loader.mixpanel_tracker import track_code_loader_loaded
-    track_code_loader_loaded()
+    track_code_loader_loaded({'event_type': 'module_import'})
 except Exception:
     pass

code_loader/leaploader.py

@@ -34,6 +34,16 @@ class LeapLoader(LeapLoaderBase):
         super().__init__(code_path, code_entry_name)
 
         self._preprocess_result_cached = None
+        
+        try:
+            from code_loader.mixpanel_tracker import track_code_loader_loaded
+            track_code_loader_loaded({
+                'event_type': 'leap_loader_instantiated',
+                'code_path': code_path,
+                'code_entry_name': code_entry_name
+            })
+        except Exception:
+            pass
 
     @lru_cache()
     def exec_script(self) -> None:

code_loader/mixpanel_tracker.py

@@ -2,8 +2,9 @@
 Mixpanel tracking utilities for code-loader.
 """
 import os
+import sys
 import getpass
-from typing import Optional
+from typing import Optional, Dict, Any
 import mixpanel
 
 
@@ -11,25 +12,18 @@ class MixpanelTracker:
     """Handles Mixpanel event tracking for code-loader."""
     
     def __init__(self, token: str = "f1bf46fb339d8c2930cde8c1acf65491"):
-        """
-        Initialize the Mixpanel tracker.
-        
-        Args:
-            token: Mixpanel project token
-        """
         self.token = token
         self.mp = mixpanel.Mixpanel(token)
-        self._user_id = None
+        self._user_id: Optional[str] = None
     
     def _get_user_id(self) -> str:
-        """Get the current user ID (whoami)."""
         if self._user_id is None:
             try:
                 self._user_id = getpass.getuser()
             except Exception:
                 # Fallback to environment variables or default
                 self._user_id = os.environ.get('USER', os.environ.get('USERNAME', 'unknown'))
-        return self._user_id
+        return self._user_id or 'unknown'
     
     def _get_tensorleap_user_id(self) -> Optional[str]:
         """Get the TensorLeap user ID from ~/.tensorleap/user_id if it exists."""
@@ -62,13 +56,7 @@ class MixpanelTracker:
             # Final fallback
             return os.environ.get('USER', os.environ.get('USERNAME', 'unknown'))
     
-    def track_code_loader_loaded(self, event_properties: Optional[dict] = None) -> None:
-        """
-        Track when code-loader is loaded.
-        
-        Args:
-            event_properties: Additional properties to include with the event
-        """
+    def track_code_loader_loaded(self, event_properties: Optional[Dict[str, Any]] = None) -> None:
         try:
             distinct_id = self._get_distinct_id()
             
@@ -77,7 +65,7 @@ class MixpanelTracker:
             
             properties = {
                 'whoami': whoami,
-                'python_version': f"{os.sys.version_info.major}.{os.sys.version_info.minor}.{os.sys.version_info.micro}",
+                'python_version': f"{sys.version_info.major}.{sys.version_info.minor}.{sys.version_info.micro}",
                 'platform': os.name,
             }
             
@@ -97,18 +85,11 @@ _tracker = None
 
 
 def get_tracker() -> MixpanelTracker:
-    """Get the global Mixpanel tracker instance."""
     global _tracker
     if _tracker is None:
         _tracker = MixpanelTracker()
     return _tracker
 
 
-def track_code_loader_loaded(event_properties: Optional[dict] = None) -> None:
-    """
-    Convenience function to track code-loader loaded event.
-    
-    Args:
-        event_properties: Additional properties to include with the event
-    """
+def track_code_loader_loaded(event_properties: Optional[Dict[str, Any]] = None) -> None:
     get_tracker().track_code_loader_loaded(event_properties)

{code_loader-1.0.114.dist-info → code_loader-1.0.116.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: code-loader
-Version: 1.0.114
+Version: 1.0.116
 Summary: 
 License: MIT
 Author: dorhar
@@ -14,7 +14,8 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: mixpanel (>=4.10.0,<5.0.0)
-Requires-Dist: numpy (>=1.22.3,<2.0.0)
+Requires-Dist: numpy (>=1.22.3,<2.0.0) ; python_version >= "3.8" and python_version < "3.11"
+Requires-Dist: numpy (>=2.3.2,<3.0.0) ; python_version >= "3.11" and python_version < "3.13"
 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)

{code_loader-1.0.114.dist-info → code_loader-1.0.116.dist-info}/RECORD

@@ -1,4 +1,4 @@
-code_loader/__init__.py,sha256=hg6VoX_Dsi1Nsmh-8R1AJ_AG9NDmWDZ03RzVVjguMWU,375
+code_loader/__init__.py,sha256=outxRQ0M-zMfV0QGVJmAed5qWfRmyD0TV6-goEGAzBw,406
 code_loader/contract/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/contract/datasetclasses.py,sha256=gJsXu4zVAaiBlq6GJwPxfTD2e0gICTtI_6Ir61MRL48,8838
 code_loader/contract/enums.py,sha256=GEFkvUMXnCNt-GOoz7NJ9ecQZ2PPDettJNOsxsiM0wk,1622
@@ -21,16 +21,16 @@ code_loader/experiment_api/workingspace_config_utils.py,sha256=DLzXQCg4dgTV_YgaS
 code_loader/inner_leap_binder/__init__.py,sha256=koOlJyMNYzGbEsoIbXathSmQ-L38N_pEXH_HvL7beXU,99
 code_loader/inner_leap_binder/leapbinder.py,sha256=0iHVHxC2NjfH7F0vQFVGy1e0llgKEyUHUHh3DdtqL70,32602
 code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=wjtk3TflrjJ8Y-OeuedVBD-09ZuOjIKGUjL7sMBU0fQ,41017
-code_loader/leaploader.py,sha256=fxf2CrWPeWfJYVoNg-Lue7RSvosuZO35FjRrDfrny5I,29363
+code_loader/leaploader.py,sha256=rQRK1lyUPSpZiRs6lKUUxVJBCe3grEY_UxEMxSIpuxI,29709
 code_loader/leaploaderbase.py,sha256=lKdw2pd6H9hFsxVmc7jJMoZd_vlG5He1ooqT-cR_yq8,4496
-code_loader/mixpanel_tracker.py,sha256=OA9d5euDCz1KSQ6mxmgmcrJsz068HzJxCB7gdSD_y20,3590
+code_loader/mixpanel_tracker.py,sha256=dxVgMug5sE99DkRonlY-Pg0l1asFH7ZtHZK_fKpk1k0,3089
 code_loader/plot_functions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/plot_functions/plot_functions.py,sha256=xg6Gi4myTN9crq6JtyrhYI38HLXjPVJcbnI7CIy8f7w,14625
 code_loader/plot_functions/visualize.py,sha256=gsBAYYkwMh7jIpJeDMPS8G4CW-pxwx6LznoQIvi4vpo,657
 code_loader/utils.py,sha256=_j8b60pimoNAvWMRj7hEkkT6C76qES6cZoBFHpXHMxA,2698
 code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/visualizers/default_visualizers.py,sha256=onRnLE_TXfgLN4o52hQIOOhUcFexGlqJ3xSpQDVLuZM,2604
-code_loader-1.0.114.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.114.dist-info/METADATA,sha256=W0sYdRjim0Swui-UxXRHn1MaiF9aS8TSxd8Il-K92Ac,955
-code_loader-1.0.114.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-code_loader-1.0.114.dist-info/RECORD,,
+code_loader-1.0.116.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-1.0.116.dist-info/METADATA,sha256=WiSm72MyFj2xP8DsMmhBN5IzzHVvYk7YVvE5KZGHs0I,1102
+code_loader-1.0.116.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+code_loader-1.0.116.dist-info/RECORD,,