code-loader 1.0.130__tar.gz → 1.0.132__tar.gz

{code_loader-1.0.130 → code_loader-1.0.132}/PKG-INFO

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

{code_loader-1.0.130 → code_loader-1.0.132}/code_loader/inner_leap_binder/leapbinder_decorators.py

@@ -92,6 +92,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 +120,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:
@@ -610,6 +628,7 @@ def tensorleap_element_instance_preprocess(
                     for instance_id in instances_ids:
                         instance_to_sample_ids_mappings[instance_id] = sample_id
                     all_sample_ids.extend(instances_ids)
+                preprocess_response.length = len(all_sample_ids)
                 preprocess_response.sample_ids_to_instance_mappings = sample_ids_to_instance_mappings
                 preprocess_response.instance_to_sample_ids_mappings = instance_to_sample_ids_mappings
                 preprocess_response.sample_ids = all_sample_ids

{code_loader-1.0.130 → code_loader-1.0.132}/pyproject.toml

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