diff --git a/benchmark/kernels/fused_moe_triton/tuning_fused_moe_triton.py b/benchmark/kernels/fused_moe_triton/tuning_fused_moe_triton.py index 937147a58..0900f8821 100644 --- a/benchmark/kernels/fused_moe_triton/tuning_fused_moe_triton.py +++ b/benchmark/kernels/fused_moe_triton/tuning_fused_moe_triton.py @@ -2,8 +2,8 @@ import argparse import json import time -from datetime import datetime from contextlib import nullcontext +from datetime import datetime from typing import Any, Dict, List, Tuple, TypedDict import ray