File tree 1 file changed +5
-5
lines changed
1 file changed +5
-5
lines changed Original file line number Diff line number Diff line change @@ -478,17 +478,17 @@ def truncate_onnx_embedding_model(
478
478
return output_filepath , tmp_file
479
479
480
480
481
- def default_cached_outputs (model : Union [ str , ModelProto ] ) -> List [bool ]:
481
+ def default_cached_outputs (model_path : str ) -> List [bool ]:
482
482
"""
483
483
Get a list of bools that indicate which outputs should be cached.
484
484
The elements that are set to True correspond to cached outputs,
485
485
the rest are set to False.
486
486
487
- :param model_path: Path to a model
487
+ :param model_path: Path to the model.
488
488
:return A list of bools that indicate which outputs should be cached.
489
489
"""
490
490
491
- outputs = get_output_names (model_path )
492
- assert len (outputs ) > 0
491
+ output_names = get_output_names (model_path )
492
+ assert len (output_names ) > 0
493
493
494
- return [output . name .startswith (CACHE_OUTPUT_PREFIX ) for output in outputs ]
494
+ return [name .startswith (CACHE_OUTPUT_PREFIX ) for name in output_names ]
You can’t perform that action at this time.
0 commit comments