Fix pytype errors.
diff --git a/compiler_opt/tools/tflite_to_cpp_lib.py b/compiler_opt/tools/tflite_to_cpp_lib.py
index 55b5799..964e314 100644
--- a/compiler_opt/tools/tflite_to_cpp_lib.py
+++ b/compiler_opt/tools/tflite_to_cpp_lib.py
@@ -158,7 +158,7 @@
def format_model(model: EmitCModel, clang_format_path: str,
- clang_format_style: str) -> str:
+ clang_format_style: str) -> EmitCModel:
"""Formats the given model and returns the result"""
logging.info('Formatting the resulting model with style [%s].',
clang_format_style)