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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -406,23 +406,12 @@ def tensorleap_load_model(prediction_types: Optional[List[PredictionTypeHandler]
406
406
  coerced = {}
407
407
  meta = {i.name: i for i in self.model.get_inputs()}
408
408
 
409
- unexpected = [n for n in float_arrays_inputs.keys() if n not in meta]
410
- if unexpected:
411
- raise ValueError(
412
- f"Input name(s) {sorted(unexpected)} do not match any ONNX model input. "
413
- f"Expected input names: {[i.name for i in self.model.get_inputs()]}"
414
- )
415
-
416
- onnx_input_order = [i.name for i in self.model.get_inputs()]
417
- provided_order = list(float_arrays_inputs.keys())
418
- if provided_order != onnx_input_order:
419
- raise ValueError(
420
- f"Input order mismatch. Provided: {provided_order}, "
421
- f"expected ONNX order: {onnx_input_order}. "
422
- f"Reorder your input functions to match the ONNX model's input order."
423
- )
424
-
425
409
  for name, arr in float_arrays_inputs.items():
410
+ if name not in meta:
411
+ # Keep as-is unless extra inputs are disallowed
412
+ coerced[name] = arr
413
+ continue
414
+
426
415
  info = meta[name]
427
416
  onnx_type = info.type
428
417
  want_dtype = ONNX_TYPE_TO_NP.get(onnx_type)
@@ -1400,7 +1389,11 @@ def tensorleap_input_encoder(name: str, channel_dim=_UNSET, model_input_index=No
1400
1389
 
1401
1390
  def mapping_inner(*args, **kwargs):
1402
1391
  class TempMapping:
1403
- pass
1392
+ def __getitem__(self, key):
1393
+ return self
1394
+
1395
+ def __iter__(self):
1396
+ return iter([])
1404
1397
 
1405
1398
  ret = TempMapping()
1406
1399
  ret.node_mapping = mapping_inner.node_mapping
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: code-loader
3
- Version: 1.0.166.dev1
3
+ Version: 1.0.166.dev2
4
4
  Summary:
5
5
  Home-page: https://github.com/tensorleap/code-loader
6
6
  License: MIT
@@ -21,7 +21,7 @@ code_loader/experiment_api/utils.py,sha256=XZHtxge12TS4H4-8PjV3sKuhp8Ud6ojAiIzTZ
21
21
  code_loader/experiment_api/workingspace_config_utils.py,sha256=DLzXQCg4dgTV_YgaSbeTVzq-2ja_SQw4zi7LXwKL9cY,990
22
22
  code_loader/inner_leap_binder/__init__.py,sha256=koOlJyMNYzGbEsoIbXathSmQ-L38N_pEXH_HvL7beXU,99
23
23
  code_loader/inner_leap_binder/leapbinder.py,sha256=Wu8Mgltf46YkqNczmYhpY-7aYIbQwTJ87hEDoyo_Mzo,33333
24
- code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=9ry0rqdZqhOutoLdlBYSFkrAtmWj7t8nmGQTumWyVt8,85462
24
+ code_loader/inner_leap_binder/leapbinder_decorators.py,sha256=ebGu60BP-cOJ0OtR9VS6bJqamHn1oQbaY0OzGjkb0KE,84871
25
25
  code_loader/leaploader.py,sha256=TW82exQGqEk36XOLonjOfx4zU4P0BoOQLbft687uU_0,30707
26
26
  code_loader/leaploaderbase.py,sha256=NXCDIIF7-ziGJccKIE9NszMSYKEE-3bn4Z4Xa3oYYOc,5909
27
27
  code_loader/mixpanel_tracker.py,sha256=eKvymkw7X2Ht6iw-a0V9VQm6OnB9kW7hYy35YtwRAvU,8457
@@ -31,7 +31,7 @@ code_loader/plot_functions/visualize.py,sha256=gsBAYYkwMh7jIpJeDMPS8G4CW-pxwx6Lz
31
31
  code_loader/utils.py,sha256=YecipkdTA-VcE9F0RQcY9cFnY8P3AksPnHM2Db7xUSk,3972
32
32
  code_loader/visualizers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
33
33
  code_loader/visualizers/default_visualizers.py,sha256=onRnLE_TXfgLN4o52hQIOOhUcFexGlqJ3xSpQDVLuZM,2604
34
- code_loader-1.0.166.dev1.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
35
- code_loader-1.0.166.dev1.dist-info/METADATA,sha256=OoMcX5862Mm_3L28NFqW-3pLpJeAn5hQxRUXOJBCcQU,1095
36
- code_loader-1.0.166.dev1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
37
- code_loader-1.0.166.dev1.dist-info/RECORD,,
34
+ code_loader-1.0.166.dev2.dist-info/LICENSE,sha256=qIwWjdspQeSMTtnFZBC8MuT-95L02FPvzRUdWFxrwJY,1067
35
+ code_loader-1.0.166.dev2.dist-info/METADATA,sha256=IvDZDCh88R7t_RnIUyUueF5s8tbwzZ88HhNS3YIHei8,1095
36
+ code_loader-1.0.166.dev2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
37
+ code_loader-1.0.166.dev2.dist-info/RECORD,,