code-loader 1.0.166.dev0__py3-none-any.whl → 1.0.166.dev2__py3-none-any.whl

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.dist-info → code_loader-1.0.166.dev2.dist-info}/METADATA

@@ -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.dist-info → code_loader-1.0.166.dev2.dist-info}/RECORD

@@ -21,7 +21,7 @@ code_loader/experiment_api/utils.py,sha256=XZHtxge12TS4H4-8PjV3sKuhp8Ud6ojAiIzTZ
 code_loader/experiment_api/workingspace_config_utils.py,sha256=DLzXQCg4dgTV_YgaSbeTVzq-2ja_SQw4zi7LXwKL9cY,990
 code_loader/inner_leap_binder/__init__.py,sha256=koOlJyMNYzGbEsoIbXathSmQ-L38N_pEXH_HvL7beXU,99
 code_loader/inner_leap_binder/leapbinder.py,sha256=Wu8Mgltf46YkqNczmYhpY-7aYIbQwTJ87hEDoyo_Mzo,33333
-code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=Hsprx732Qc_XU-QasDsajAX0EpNdkHvdYB_btBCund4,84929
+code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=ebGu60BP-cOJ0OtR9VS6bJqamHn1oQbaY0OzGjkb0KE,84871
 code_loader/leaploader.py,sha256=TW82exQGqEk36XOLonjOfx4zU4P0BoOQLbft687uU_0,30707
 code_loader/leaploaderbase.py,sha256=NXCDIIF7-ziGJccKIE9NszMSYKEE-3bn4Z4Xa3oYYOc,5909
 code_loader/mixpanel_tracker.py,sha256=eKvymkw7X2Ht6iw-a0V9VQm6OnB9kW7hYy35YtwRAvU,8457
@@ -31,7 +31,7 @@ code_loader/plot_functions/visualize.py,sha256=gsBAYYkwMh7jIpJeDMPS8G4CW-pxwx6Lz
 code_loader/utils.py,sha256=YecipkdTA-VcE9F0RQcY9cFnY8P3AksPnHM2Db7xUSk,3972
 code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 code_loader/visualizers/default_visualizers.py,sha256=onRnLE_TXfgLN4o52hQIOOhUcFexGlqJ3xSpQDVLuZM,2604
-code_loader-1.0.166.dev0.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
-code_loader-1.0.166.dev0.dist-info/METADATA,sha256=YdKmWgzeWArN9u2_DEZZJ-g2qJQ4rnR-UvUcwuamGjM,1095
-code_loader-1.0.166.dev0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-code_loader-1.0.166.dev0.dist-info/RECORD,,
+code_loader-1.0.166.dev2.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
+code_loader-1.0.166.dev2.dist-info/METADATA,sha256=IvDZDCh88R7t_RnIUyUueF5s8tbwzZ88HhNS3YIHei8,1095
+code_loader-1.0.166.dev2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+code_loader-1.0.166.dev2.dist-info/RECORD,,