fix issues

Author: xadupre

onnx_diagnostic/helpers/helper.py

@@ -872,7 +872,7 @@ def max_diff(
     _index: int = 0,
     allow_unique_tensor_with_list_of_one_element: bool = True,
     hist: Optional[Union[bool, List[float]]] = None,
-) -> Dict[str, Union[float, str, int, Tuple[int, ...]]]:
+) -> Dict[str, Union[float, int, Tuple[int, ...]]]:
     """
     Returns the maximum discrepancy.
 
@@ -1221,7 +1221,7 @@ def max_diff(
                     f"_index={_index}"
                 )
 
-        res: Dict[str, Union[str, int, float, Tuple[int, ...]]] = dict(
+        res: Dict[str, Union[int, float, Tuple[int, ...]]] = dict(
             abs=abs_diff, rel=rel_diff, sum=sum_diff, n=n_diff, dnan=nan_diff, argm=argm
         )
         if hist:
@@ -1332,7 +1332,7 @@ def max_diff(
                     f"_index={_index}"
                 )
 
-        res: Dict[str, Union[str, int, float, Tuple[int, ...]]] = dict(
+        res: Dict[str, Union[int, float, Tuple[int, ...]]] = dict(
             abs=abs_diff, rel=rel_diff, sum=sum_diff, n=n_diff, dnan=nan_diff, argm=argm
         )
         if hist: