code-loader 1.0.131__tar.gz → 1.0.132.dev1__tar.gz

{code_loader-1.0.131 → code_loader-1.0.132.dev1}/PKG-INFO

@@ -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 → code_loader-1.0.132.dev1}/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-1.0.131 → code_loader-1.0.132.dev1}/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-1.0.131 → code_loader-1.0.132.dev1}/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 → code_loader-1.0.132.dev1}/pyproject.toml

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