From 94c7c3be2c1f731bb1f27412cc52ba355d6ce373 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Haian=20Huang=28=E6=B7=B1=E5=BA=A6=E7=9C=B8=29?=
<1286304229@qq.com>
Date: Mon, 6 Jun 2022 14:02:15 +0800
Subject: [PATCH] [Enhance]: remove warning in vis_backend (#273)
* fix visbackend warning
* fix
---
mmengine/visualization/vis_backend.py | 12 +++++++-----
1 file changed, 7 insertions(+), 5 deletions(-)
diff --git a/mmengine/visualization/vis_backend.py b/mmengine/visualization/vis_backend.py
index fb1b6cae..2c1dedc9 100644
--- a/mmengine/visualization/vis_backend.py
+++ b/mmengine/visualization/vis_backend.py
@@ -12,6 +12,7 @@ import torch
from mmengine.config import Config
from mmengine.fileio import dump
+from mmengine.logging import MMLogger
from mmengine.registry import VISBACKENDS
from mmengine.utils import TORCH_VERSION
@@ -40,11 +41,12 @@ def force_init_env(old_func: Callable) -> Any:
# `_env_initialized` is False, call `_init_env` and set
# `_env_initialized` to True
if not getattr(obj, '_env_initialized', False):
- warnings.warn('Attribute `_env_initialized` is not defined in '
- f'{type(obj)} or `type(obj)._env_initialized is '
- 'False, `_init_env` will be called and '
- f'{type(obj)}._env_initialized will be set to '
- 'True')
+ logger = MMLogger.get_current_instance()
+ logger.debug('Attribute `_env_initialized` is not defined in '
+ f'{type(obj)} or `{type(obj)}._env_initialized is '
+ 'False, `_init_env` will be called and '
+ f'{type(obj)}._env_initialized will be set to '
+ 'True')
obj._init_env() # type: ignore
obj._env_initialized = True # type: ignore
--
GitLab