fix

Signed-off-by: Jinzhe Zeng <[email protected]>

deepmd/loggers.py → deepmd/loggers/__init__.py

@@ -1,3 +1,8 @@
 # SPDX-License-Identifier: LGPL-3.0-or-later
 """Alias of deepmd_utils.loggers for backward compatibility."""
-from deepmd_utils.loggers import *  # noqa: F403
+
+from deepmd_utils.loggers.loggers import (
+    set_log_handles,
+)
+
+__all__ = ["set_log_handles"]

deepmd/loggers/loggers.py

@@ -0,0 +1,3 @@
+# SPDX-License-Identifier: LGPL-3.0-or-later
+"""Alias of deepmd_utils.loggers.loggers for backward compatibility."""
+from deepmd_utils.loggers.loggers import *  # noqa: F403

deepmd/model/model_stat.py

@@ -1,3 +1,13 @@
 # SPDX-License-Identifier: LGPL-3.0-or-later
 """Alias for backward compatibility."""
-from deepmd_utils.utils.model_stat import *  # noqa: F403
+from deepmd_utils.utils.model_stat import (
+    _make_all_stat_ref,
+    make_stat_input,
+    merge_sys_stat,
+)
+
+__all__ = [
+    "make_stat_input",
+    "merge_sys_stat",
+    "_make_all_stat_ref",  # used by tests
+]