Fix linting & rebase.

python/tvm/meta_schedule/cost_model/cost_model.py

@@ -208,4 +208,4 @@ def __str__(self) -> str:
         result : str
             Get the cost model as string with name.
         """
-        return f"meta_schedule.{self.__class__.__name__}({_get_hex_address(self.handle)})"
+        return f"meta_schedule.{self.__class__.__name__}({_get_hex_address(self.handle)})"  # type: ignore

python/tvm/meta_schedule/cost_model/random_model.py

@@ -19,7 +19,6 @@
 """
 from typing import List, Optional, Tuple, Union
 
-import numpy as np
 from tvm.meta_schedule.utils import derived_object  # type: ignore
 
 from ..cost_model import PyCostModel
@@ -46,6 +45,8 @@ class RandomModel(PyCostModel):
     https://numpy.org/doc/stable/reference/random/generated/numpy.random.get_state.html
     """
 
+    import numpy as np  # pylint: disable=import-outside-toplevel
+
     random_state: Union[Tuple[str, np.ndarray, int, int, float], dict]
     path: Optional[str]
 
@@ -56,6 +57,8 @@ def __init__(
         path: Optional[str] = None,
         max_range: Optional[int] = 100,
     ):
+        import numpy as np  # pylint: disable=import-outside-toplevel
+
         super().__init__()
         if path is not None:
             self.load(path)
@@ -72,6 +75,8 @@ def load(self, path: str) -> None:
         path : str
             The file path.
         """
+        import numpy as np  # pylint: disable=import-outside-toplevel
+
         self.random_state = tuple(np.load(path, allow_pickle=True))  # type: ignore
 
     def save(self, path: str) -> None:
@@ -82,6 +87,8 @@ def save(self, path: str) -> None:
         path : str
             The file path.
         """
+        import numpy as np  # pylint: disable=import-outside-toplevel
+
         np.save(path, np.array(self.random_state, dtype=object), allow_pickle=True)
 
     def update(
@@ -117,6 +124,8 @@ def predict(self, context: TuneContext, candidates: List[MeasureCandidate]) -> n
         result : np.ndarray
             The predicted running results.
         """
+        import numpy as np  # pylint: disable=import-outside-toplevel
+
         np.random.set_state(self.random_state)
         # TODO(@zxybazh): Use numpy's RandState object:
         # https://numpy.org/doc/1.16/reference/generated/numpy.random.RandomState.html#numpy.random.RandomState

python/tvm/meta_schedule/feature_extractor/feature_extractor.py

@@ -105,4 +105,4 @@ def extract_from(
         raise NotImplementedError
 
     def __str__(self) -> str:
-        return f"meta_schedule.{self.__class__.__name__}({_get_hex_address(self.handle)})"
+        return f"meta_schedule.{self.__class__.__name__}({_get_hex_address(self.handle)})"  # type: ignore

python/tvm/meta_schedule/measure_callback/measure_callback.py

@@ -125,4 +125,4 @@ def apply(
         raise NotImplementedError
 
     def __str__(self) -> str:
-        return f"meta_schedule.{self.__class__.__name__}({_get_hex_address(self.handle)})"
+        return f"meta_schedule.{self.__class__.__name__}({_get_hex_address(self.handle)})"  # type: ignore

python/tvm/meta_schedule/mutator/mutator.py

@@ -122,4 +122,4 @@ def __str__(self) -> str:
         result : str
             Get the mutator as string with name.
         """
-        return f"meta_schedule.{self.__class__.__name__}({_get_hex_address(self.handle)})"
+        return f"meta_schedule.{self.__class__.__name__}({_get_hex_address(self.handle)})"  # type: ignore

python/tvm/meta_schedule/postproc/postproc.py

@@ -124,4 +124,4 @@ def __str__(self) -> str:
         result : str
             Get the post processor as string with name.
         """
-        return f"meta_schedule.{self.__class__.__name__}({_get_hex_address(self.handle)})"
+        return f"meta_schedule.{self.__class__.__name__}({_get_hex_address(self.handle)})"  # type: ignore

python/tvm/meta_schedule/schedule_rule/schedule_rule.py

@@ -137,4 +137,4 @@ def __str__(self) -> str:
         result : str
             Get the schedule rule as string with name.
         """
-        return f"meta_schedule.{self.__class__.__name__}({_get_hex_address(self.handle)})"
+        return f"meta_schedule.{self.__class__.__name__}({_get_hex_address(self.handle)})"  # type: ignore

python/tvm/meta_schedule/tune.py

@@ -291,7 +291,7 @@ def _callbacks(
     @staticmethod
     def _cost_model(cost_model: Optional[CostModel]) -> CostModel:
         if cost_model is None:
-            return XGBModel(extractor=PerStoreFeature())
+            return XGBModel(extractor=PerStoreFeature())  # type: ignore
         if not isinstance(cost_model, CostModel):
             raise TypeError(f"Expected `cost_model` to be CostModel, but gets: {cost_model}")
         return cost_model

python/tvm/meta_schedule/utils.py

@@ -125,7 +125,7 @@ def __setattr__(self, name, value):
             else:
                 super(TVMDerivedObject, self).__setattr__(name, value)
 
-    functools.update_wrapper(TVMDerivedObject.__init__, cls.__init__)
+    functools.update_wrapper(TVMDerivedObject.__init__, cls.__init__)  # type: ignore
     TVMDerivedObject.__name__ = cls.__name__
     TVMDerivedObject.__doc__ = cls.__doc__
     TVMDerivedObject.__module__ = cls.__module__