openxla / stablehlo

Backward compatible ML compute opset inspired by HLO/MHLO
Apache License 2.0
357 stars 100 forks source link

Fix the return type of python generator function #2378

Closed sdasgup3 closed 1 month ago

sdasgup3 commented 1 month ago

The return type of 'List[tensorflow.TensorSpec]' for generator function StableHLOToTFSavedModel._make_input_signatures is error-ed out during internal testing.

The fix is to annotating the return type as Iterator as all generators are basically iterators.