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

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

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

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

@@ -89,9 +89,42 @@ def tensorleap_load_model(prediction_types: Optional[List[PredictionTypeHandler]
                     ret = self.model(arg)
                     return ret.numpy()
 
+                def _convert_onnx_inputs_to_correct_type(
+                        self, float_arrays_inputs: Dict[str, np.ndarray]
+                ) -> Dict[str, np.ndarray]:
+                    """
+                    Cast user-provided NumPy inputs to match the dtypes/shapes
+                    expected by an ONNX Runtime InferenceSession.
+                    """
+                    coerced = {}
+                    meta = {i.name: i for i in self.model.get_inputs()}
+
+                    for name, arr in float_arrays_inputs.items():
+                        if name not in meta:
+                            # Keep as-is unless extra inputs are disallowed
+                            coerced[name] = arr
+                            continue
+
+                        info = meta[name]
+                        want_dtype = np.dtype(info.type)  # ONNX Runtime gives NumPy dtype string
+
+                        # Cast dtype if needed
+                        if arr.dtype != want_dtype:
+                            arr = arr.astype(want_dtype, copy=False)
+
+                        coerced[name] = arr
+
+                    # Verify required inputs are present
+                    missing = [n for n in meta if n not in coerced]
+                    if missing:
+                        raise KeyError(f"Missing required input(s): {sorted(missing)}")
+
+                    return coerced
+
                 # onnx runtime interface
                 def run(self, output_names, input_dict):
-                    return self.model.run(output_names, input_dict)
+                    corrected_type_inputs = self._convert_onnx_inputs_to_correct_type(input_dict)
+                    return self.model.run(output_names, corrected_type_inputs)
 
                 def get_inputs(self):
                     return self.model.get_inputs()

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

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