mypy

Author: xadupre

onnx_diagnostic/_command_lines_parser.py

@@ -768,12 +768,8 @@ def get_parser_agg() -> ArgumentParser:
 
 
 def _cmd_agg(argv: List[Any]):
-    from .helpers.log_helper import (
-        CubeLogsPerformance,
-        open_dataframe,
-        enumerate_csv_files,
-        filter_data,
-    )
+    from .helpers._log_helper import CubeLogsPerformance
+    from .helpers.log_helper import open_dataframe, enumerate_csv_files, filter_data
 
     parser = get_parser_agg()
     args = parser.parse_args(argv[1:])

onnx_diagnostic/helpers/_log_helper.py

@@ -316,7 +316,7 @@ def align_dataframe_with(
 
 def apply_excel_style(
     filename_or_writer: Any,
-    f_highlights: Optional[
+    f_highlights: Optional[  # type: ignore[name-defined]
         Dict[str, Callable[[Any], "CubeViewDef.HighLightKind"]]  # noqa: F821
     ] = None,
     time_mask_view: Optional[Dict[str, pandas.DataFrame]] = None,