code-loader 1.0.166.dev0__tar.gz → 1.0.166.dev1__tar.gz

{code_loader-1.0.166.dev0 → code_loader-1.0.166.dev1}/PKG-INFO

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

{code_loader-1.0.166.dev0 → code_loader-1.0.166.dev1}/code_loader/inner_leap_binder/leapbinder_decorators.py

@@ -413,6 +413,15 @@ def tensorleap_load_model(prediction_types: Optional[List[PredictionTypeHandler]
                             f"Expected input names: {[i.name for i in self.model.get_inputs()]}"
                         )
 
+                    onnx_input_order = [i.name for i in self.model.get_inputs()]
+                    provided_order = list(float_arrays_inputs.keys())
+                    if provided_order != onnx_input_order:
+                        raise ValueError(
+                            f"Input order mismatch. Provided: {provided_order}, "
+                            f"expected ONNX order: {onnx_input_order}. "
+                            f"Reorder your input functions to match the ONNX model's input order."
+                        )
+
                     for name, arr in float_arrays_inputs.items():
                         info = meta[name]
                         onnx_type = info.type

{code_loader-1.0.166.dev0 → code_loader-1.0.166.dev1}/pyproject.toml

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