Update tools/test.py

open-mmlab · Jun 2, 2024 · 4512c03 · 4512c03
1 parent 17deea9
commit 4512c03
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/test.py b/tools/test.py
@@ -122,7 +122,7 @@ def main():
             cfg.test_evaluator = list(cfg.test_evaluator)
             cfg.test_evaluator.append(dump_metric)
         elif isinstance(cfg.test_evaluator, dict):
-            if 'metrics' in cfg.test_evaluator.keys():
+            if 'metrics' in cfg.test_evaluator:
                 cfg.test_evaluator['metrics'].append(dump_metric)
             else:
                 cfg.test_evaluator['metrics'] = [dump_metric]