fix ut

open-mmlab · Nov 10, 2022 · ab6df83 · ab6df83
1 parent 94e0a7c
commit ab6df83
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_runners/test_evolution_search_loop.py b/tests/test_runners/test_evolution_search_loop.py
@@ -359,7 +359,7 @@ def test_run_loop(self, mock_flops, mock_model2vector,
         mock_flops.return_value = (0.5, 101)
         mock_export_fix_subnet.return_value = fake_subnet
         mock_model2vector.return_value = dict(
-            feats=[0, 1], onehot_feats=[0, 1, 0, 1])
+            normal_vector=[0, 1], onehot_vector=[0, 1, 0, 1])
 
         loop.run()
         assert os.path.exists(