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

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

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

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

@@ -406,14 +406,12 @@ def tensorleap_load_model(prediction_types: Optional[List[PredictionTypeHandler]
                     coerced = {}
                     meta = {i.name: i for i in self.model.get_inputs()}
 
-                    unexpected = [n for n in float_arrays_inputs.keys() if n not in meta]
-                    if unexpected:
-                        raise ValueError(
-                            f"Input name(s) {sorted(unexpected)} do not match any ONNX model input. "
-                            f"Expected input names: {[i.name 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]
                         onnx_type = info.type
                         want_dtype = ONNX_TYPE_TO_NP.get(onnx_type)
@@ -1391,7 +1389,11 @@ def tensorleap_input_encoder(name: str, channel_dim=_UNSET, model_input_index=No
 
         def mapping_inner(*args, **kwargs):
             class TempMapping:
-                pass
+                def __getitem__(self, key):
+                    return self
+
+                def __iter__(self):
+                    return iter([])
 
             ret = TempMapping()
             ret.node_mapping = mapping_inner.node_mapping

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

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "code-loader"
-version = "1.0.166.dev0"
+version = "1.0.166.dev2"
 description = ""
 authors = ["dorhar <doron.harnoy@tensorleap.ai>"]
 license = "MIT"
@@ -28,8 +28,6 @@ grappa = "^1.0.1"
 pytest-cov = "^3.0.0"
 pytest-watch = "^4.2.0"
 mypy = "1.3.0"
-onnxruntime = {version = ">=1.0.0", python = ">=3.10"}
-onnx = {version = ">=1.0.0", python = ">=3.10"}
 
 [build-system]
 requires = ["poetry-core>=1.0.0"]