code-loader 1.0.131__py3-none-any.whl → 1.0.132.dev1__py3-none-any.whl

code_loader/contract/mapping.py

@@ -21,12 +21,22 @@ class NodeMappingType(Enum):
     Prediction1 = 'Prediction1'
     Prediction2 = 'Prediction2'
     Prediction3 = 'Prediction3'
+    Prediction4 = 'Prediction4'
+    Prediction5 = 'Prediction5'
+    Prediction6 = 'Prediction6'
+    Prediction7 = 'Prediction7'
+    Prediction8 = 'Prediction8'
+    Prediction9 = 'Prediction9'
     Input0 = 'Input0'
     Input1 = 'Input1'
     Input2 = 'Input2'
     Input3 = 'Input3'
     Input4 = 'Input4'
     Input5 = 'Input5'
+    Input6 = 'Input6'
+    Input7 = 'Input7'
+    Input8 = 'Input8'
+    Input9 = 'Input9'
     PredictionLabels = 'PredictionLabels'
 
 
@@ -39,6 +49,7 @@ class NodeMapping:
     arg_names: Optional[List[str]] = None
 
 
+
 @dataclass
 class NodeConnection:
     node: NodeMapping

code_loader/inner_leap_binder/leapbinder_decorators.py

@@ -69,6 +69,11 @@ def integration_test():
 
     return decorating_function
 
+def _safe_get_item(key):
+    try:
+        return NodeMappingType[f'Input{str(key)}']
+    except ValueError:
+        raise Exception(f'Tensorleap currently supports models with no more then 10 inputs')
 
 def tensorleap_load_model(prediction_types: Optional[List[PredictionTypeHandler]] = None):
     for i, prediction_type in enumerate(prediction_types):
@@ -92,6 +97,20 @@ def tensorleap_load_model(prediction_types: Optional[List[PredictionTypeHandler]
                 def _convert_onnx_inputs_to_correct_type(
                         self, float_arrays_inputs: Dict[str, np.ndarray]
                 ) -> Dict[str, np.ndarray]:
+                    ONNX_TYPE_TO_NP = {
+                        "tensor(float)": np.float32,
+                        "tensor(double)": np.float64,
+                        "tensor(int64)": np.int64,
+                        "tensor(int32)": np.int32,
+                        "tensor(int16)": np.int16,
+                        "tensor(int8)": np.int8,
+                        "tensor(uint64)": np.uint64,
+                        "tensor(uint32)": np.uint32,
+                        "tensor(uint16)": np.uint16,
+                        "tensor(uint8)": np.uint8,
+                        "tensor(bool)": np.bool_,
+                    }
+
                     """
                     Cast user-provided NumPy inputs to match the dtypes/shapes
                     expected by an ONNX Runtime InferenceSession.
@@ -106,7 +125,11 @@ def tensorleap_load_model(prediction_types: Optional[List[PredictionTypeHandler]
                             continue
 
                         info = meta[name]
-                        want_dtype = np.dtype(info.type)  # ONNX Runtime gives NumPy dtype string
+                        onnx_type = info.type
+                        want_dtype = ONNX_TYPE_TO_NP.get(onnx_type)
+
+                        if want_dtype is None:
+                            raise TypeError(f"Unsupported ONNX input type: {onnx_type}")
 
                         # Cast dtype if needed
                         if arr.dtype != want_dtype:
@@ -141,15 +164,20 @@ def tensorleap_load_model(prediction_types: Optional[List[PredictionTypeHandler]
                         f'Expected key to be an int, got {type(key)} instead.'
 
                     ret = TempMapping()
-                    ret.node_mapping = NodeMapping('', NodeMappingType(f'Prediction{str(key)}'))
+                    try:
+                        ret.node_mapping = NodeMapping('', NodeMappingType(f'Prediction{str(key)}'))
+                    except ValueError as e:
+                        raise Exception(f'Tensorleap currently supports models with no more then 10 active predictions,'
+                                        f' {key} not supported.')
                     return ret
 
             class ModelPlaceholder:
+
                 # keras interface
                 def __call__(self, arg):
                     if isinstance(arg, list):
                         for i, elem in enumerate(arg):
-                            elem.node_mapping.type = NodeMappingType[f'Input{str(i)}']
+                            elem.node_mapping.type = _safe_get_item(i)
                     else:
                         arg.node_mapping.type = NodeMappingType.Input0
 
@@ -164,14 +192,14 @@ def tensorleap_load_model(prediction_types: Optional[List[PredictionTypeHandler]
                         if isinstance(input_key, NodeMappingType):
                             elem.node_mapping.type = input_key
                         else:
-                            elem.node_mapping.type = NodeMappingType[f'Input{str(i)}']
+                            elem.node_mapping.type = _safe_get_item(i)
 
                     return ModelOutputPlaceholder()
 
                 def get_inputs(self):
                     class FollowIndex:
                         def __init__(self, index):
-                            self.name =  NodeMappingType[f'Input{str(i)}']
+                            self.name =  _safe_get_item(index)
 
                     class FollowInputIndex:
                         def __init__(self):

code_loader/mixpanel_tracker.py

@@ -8,11 +8,13 @@ import uuid
 from typing import Optional, Dict, Any
 import mixpanel  # type: ignore[import]
 
+TRACKING_VERSION = '1'
+
 
 class MixpanelTracker:
     """Handles Mixpanel event tracking for code-loader."""
     
-    def __init__(self, token: str = "f1bf46fb339d8c2930cde8c1acf65491"):
+    def __init__(self, token: str = "0c1710c9656bbfb1056bb46093e23ca1"):
         self.token = token
         self.mp = mixpanel.Mixpanel(token)
         self._user_id: Optional[str] = None
@@ -106,6 +108,8 @@ class MixpanelTracker:
             device_id = self._get_or_create_device_id()
             
             properties = {
+                'tracking_version': TRACKING_VERSION,
+                'service': 'code-loader',
                 'whoami': whoami,
                 '$device_id': device_id,  # Always use device_id for $device_id
                 'python_version': f"{sys.version_info.major}.{sys.version_info.minor}.{sys.version_info.micro}",

{code_loader-1.0.131.dist-info → code_loader-1.0.132.dev1.dist-info}/METADATA

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

{code_loader-1.0.131.dist-info → code_loader-1.0.132.dev1.dist-info}/RECORD

@@ -4,7 +4,7 @@ code_loader/contract/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hS
 code_loader/contract/datasetclasses.py,sha256=u0gfDDy02skhFG3ejJOxqxCnykhAcBPGJfv8Bi4s9eQ,8966
 code_loader/contract/enums.py,sha256=GEFkvUMXnCNt-GOoz7NJ9ecQZ2PPDettJNOsxsiM0wk,1622
 code_loader/contract/exceptions.py,sha256=jWqu5i7t-0IG0jGRsKF4DjJdrsdpJjIYpUkN1F4RiyQ,51
-code_loader/contract/mapping.py,sha256=e11h_sprwOyE32PcqgRq9JvyahQrPzwqgkhmbQLKLQY,1165
+code_loader/contract/mapping.py,sha256=sWJhpng-IkOzQnWQdMT5w2ZZ3X1Z_OOzSwCLXIS7oxE,1446
 code_loader/contract/responsedataclasses.py,sha256=6-5DJkYBdXb3UB1eNidTTPPBIYxMjEoMdYDkp9VhH8o,4223
 code_loader/contract/visualizer_classes.py,sha256=Wz9eItmoRaKEHa3p0aW0Ypxx4_xUmaZyLBznnTuxwi0,15425
 code_loader/default_losses.py,sha256=NoOQym1106bDN5dcIk56Elr7ZG5quUHArqfP5-Nyxyo,1139
@@ -21,17 +21,17 @@ code_loader/experiment_api/utils.py,sha256=XZHtxge12TS4H4-8PjV3sKuhp8Ud6ojAiIzTZ
 code_loader/experiment_api/workingspace_config_utils.py,sha256=DLzXQCg4dgTV_YgaSbeTVzq-2ja_SQw4zi7LXwKL9cY,990
 code_loader/inner_leap_binder/__init__.py,sha256=koOlJyMNYzGbEsoIbXathSmQ-L38N_pEXH_HvL7beXU,99
 code_loader/inner_leap_binder/leapbinder.py,sha256=Q3D9yVM-GNEJfYRFvMV__BoZbcWOgnWKhrZXAv6Tu7o,33232
-code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=wXZT-wNvixOC3-iKOf1YMWbr0DWQgSA5WsS2UetQ90k,48693
+code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=WnHy5tdSwPI3RRFMcIHG_SYw0vaqE7t5URyhEtO635E,49896
 code_loader/leaploader.py,sha256=85XUWd7Y8kup76xeqefgd8db7pGaeHMhE7QlmYlNFMw,29747
 code_loader/leaploaderbase.py,sha256=LIFcC6xo6V_iiGN3BjibXETu_l84EWM_WIOKAvkfTiM,4458
-code_loader/mixpanel_tracker.py,sha256=mJaJvs8Pc5w3FEmSSObFIMVekcs5pKdM3iZmN4wVFqA,4822
+code_loader/mixpanel_tracker.py,sha256=l9z_szKKQ7apEbdNZpGH1TKAiT_TsBHb9AQnePaWTyo,4942
 code_loader/plot_functions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/plot_functions/plot_functions.py,sha256=VyVWxd7R3lALIo2z8oZlYybbN0Ip6G0OiKNTNZ77xHk,14557
 code_loader/plot_functions/visualize.py,sha256=gsBAYYkwMh7jIpJeDMPS8G4CW-pxwx6LznoQIvi4vpo,657
 code_loader/utils.py,sha256=gXENTYpjdidq2dx0gVbXlErPeHoNs-4TYAZbLRe0y2c,2712
 code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/visualizers/default_visualizers.py,sha256=onRnLE_TXfgLN4o52hQIOOhUcFexGlqJ3xSpQDVLuZM,2604
-code_loader-1.0.131.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.131.dist-info/METADATA,sha256=J6wQd25HCrreLLrxFsLGpQN-rskZXT2jqeTOhJBcyQM,1090
-code_loader-1.0.131.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-code_loader-1.0.131.dist-info/RECORD,,
+code_loader-1.0.132.dev1.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-1.0.132.dev1.dist-info/METADATA,sha256=Ay8oSYnVpGPGak8xrsh8k4L7Yv5yM8cn4_l_uqPUpLU,1095
+code_loader-1.0.132.dev1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+code_loader-1.0.132.dev1.dist-info/RECORD,,