diff --git a/python/tvm/meta_schedule/__init__.py b/python/tvm/meta_schedule/__init__.py index 336a737cdd19b..b5ad51329ad92 100644 --- a/python/tvm/meta_schedule/__init__.py +++ b/python/tvm/meta_schedule/__init__.py @@ -36,4 +36,3 @@ ) from .tune import tune_te, tune_tir, tune_relay from .tune_context import TuneContext -from .tune import tune_tir, tune_te diff --git a/tests/python/unittest/test_meta_schedule_search_strategy.py b/tests/python/unittest/test_meta_schedule_search_strategy.py index d0cdc3571bd7b..5d816348f48a0 100644 --- a/tests/python/unittest/test_meta_schedule_search_strategy.py +++ b/tests/python/unittest/test_meta_schedule_search_strategy.py @@ -269,4 +269,5 @@ def predict( if __name__ == "__main__": - sys.exit(pytest.main([__file__] + sys.argv[1:])) + test_meta_schedule_evolutionary_search() + # sys.exit(pytest.main([__file__] + sys.argv[1:]))