py36 ci

open-mmlab · Oct 17, 2022 · c314033 · c314033
1 parent ad4f9db
commit c314033
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 4 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -30,6 +30,15 @@ repos:
     rev: v2.1.0
     hooks:
       - id: codespell
+  - repo: https://github.com/executablebooks/mdformat
+    rev: 0.7.14
+    hooks:
+      - id: mdformat
+        args: ["--number"]
+        additional_dependencies:
+          - mdformat-gfm
+          - mdformat_frontmatter
+          - linkify-it-py
   - repo: https://github.com/myint/docformatter
     rev: v1.3.1
     hooks:

diff --git a/tests/test_runners/test_evolution_search_loop.py b/tests/test_runners/test_evolution_search_loop.py
@@ -183,12 +183,12 @@ def test_run_loop(self, mock_flops, mock_export_fix_subnet):
 
         mutation_candidates = Candidates([fake_subnet] * loop.num_mutation)
         for i in range(loop.num_mutation):
-            mutation_candidates.set_resources(i, 0.1 + 0.1 * i)
-            mutation_candidates.set_score(i, 99 + i)
+            mutation_candidates.set_resource(i, 0.1 + 0.1 * i, 'flops')
+            mutation_candidates.set_resource(i, 99 + i, 'score')
         crossover_candidates = Candidates([fake_subnet] * loop.num_crossover)
         for i in range(loop.num_crossover):
-            crossover_candidates.set_resources(i, 0.1 + 0.1 * i)
-            crossover_candidates.set_score(i, 99 + i)
+            crossover_candidates.set_resource(i, 0.1 + 0.1 * i, 'flops')
+            crossover_candidates.set_resource(i, 99 + i, 'score')
         loop.gen_mutation_candidates = \
             MagicMock(return_value=mutation_candidates)
         loop.gen_crossover_candidates = \