fix pytest

open-mmlab · Oct 22, 2022 · b49adc3 · b49adc3
1 parent 290fec3
commit b49adc3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/tests/test_core/test_delivers/test_function_outputs_deliver.py b/tests/test_core/test_delivers/test_function_outputs_deliver.py
@@ -142,6 +142,7 @@ def test_ema_hook(self):
                 num_workers=0),
             val_evaluator=evaluator,
             work_dir=self.temp_dir.name,
+            default_scope='mmrazor',
             optim_wrapper=OptimWrapper(
                 torch.optim.Adam(ToyModel().parameters())),
             train_cfg=dict(by_epoch=True, max_epochs=2, val_interval=1),

diff --git a/tests/test_core/test_recorders/test_func_inputs_recorder.py b/tests/test_core/test_recorders/test_func_inputs_recorder.py
@@ -117,6 +117,7 @@ def test_ema_hook(self):
                 num_workers=0),
             val_evaluator=evaluator,
             work_dir=self.temp_dir.name,
+            default_scope='mmrazor',
             optim_wrapper=OptimWrapper(
                 torch.optim.Adam(ToyModel().parameters())),
             train_cfg=dict(by_epoch=True, max_epochs=2, val_interval=1),