formatted

apache · Jun 10, 2022 · eabad87 · eabad87
1 parent 7582b97
commit eabad87
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/python/tvm/meta_schedule/testing/dataset_extract_tasks.py b/python/tvm/meta_schedule/testing/dataset_extract_tasks.py
@@ -77,7 +77,7 @@ def extract_and_save_tasks(cache_file):
     except OSError as error:
         print(f"Directory {args.task_cache_dir} cannot be created successfully.")
 
-    is_spatial = tvm.get_global_func(
+    is_spatial = tvm.get_global_func(  # pylint: disable=invalid-name
         "tir.schedule.IsSpatialPrimFunc"
     )  # pylint: disable=invalid-name
     paths = glob.glob(os.path.join(args.model_cache_dir, "*.json"))  # pylint: disable=invalid-name

diff --git a/python/tvm/meta_schedule/testing/dataset_sample_candidates.py b/python/tvm/meta_schedule/testing/dataset_sample_candidates.py
@@ -162,7 +162,7 @@ def sample_candidates(task, task_name, model_name):
     evolve_with_cost_model = tvm.get_global_func(  # pylint: disable=invalid-name
         "meta_schedule.SearchStrategyEvolutionarySearchEvolveWithCostModel"
     )
-    task_paths = sorted(
+    task_paths = sorted(  # pylint: disable=invalid-name
         glob.glob(os.path.join(args.task_cache_dir, "*.json"))
     )[  # pylint: disable=invalid-name
         args.file_group * 10 : (args.file_group + 1) * 10