fix lint again

python/tvm/meta_schedule/cost_model/xgb_model.py

@@ -23,13 +23,8 @@
 from collections import OrderedDict
 from itertools import chain as itertools_chain
 from typing import TYPE_CHECKING, Any, Callable, Dict, List, NamedTuple, Optional, Tuple
-
-# TODO: This is to suppress the FutureWarnings from older version xgboost using pandas.Int64Index
-# Should be removed after TVM requires the use of xgboost >= 1.6.0
 import warnings
 
-warnings.simplefilter(action="ignore", category=FutureWarning)
-
 import numpy as np  # type: ignore
 
 from ...contrib.tar import tar, untar
@@ -65,6 +60,9 @@ class OptXGBoostCustomCallback(cls, TrainingCallback):  # type: ignore
 
     from ..tune_context import TuneContext
 
+# TODO: This is to suppress the FutureWarnings from older version xgboost using pandas.Int64Index
+# Should be removed after TVM requires the use of xgboost >= 1.6.0
+warnings.simplefilter(action="ignore", category=FutureWarning)
 
 logger = logging.getLogger(__name__)  # pylint: disable=invalid-name