add 3.10

Author: xadupre

.github/workflows/ci.yml

@@ -15,7 +15,7 @@ jobs:
     strategy:
       matrix:
         os: [ubuntu-latest]
-        python: ['3.11', '3.12']
+        python: ['3.10', '3.11', '3.12']
         transformers: ['4.48.3', '4.51.3', 'main']
         torch: ['2.6', 'main']
 

onnx_diagnostic/tasks/mixture_of_expert.py

@@ -1,5 +1,6 @@
 from typing import Any, Callable, Dict, Optional, Tuple
 import torch
+
 # from ..helpers.cache_helper import make_dynamic_cache
 from ..helpers.config_helper import update_config  # , check_hasattr, _pick