浏览代码

fix logging

Denis Mazur 4 年之前
父节点
当前提交
ec4bb554ed
共有 1 个文件被更改,包括 2 次插入1 次删除
  1. 2 1
      benchmarks/benchmark_throughput_p2p.py

+ 2 - 1
benchmarks/benchmark_throughput_p2p.py

@@ -11,8 +11,9 @@ from hivemind import P2P
 from hivemind.dht import DHT
 from hivemind.moe.server import layers
 from hivemind.utils.limits import increase_file_limit
-from hivemind.utils.logging import get_logger
+from hivemind.utils.logging import get_logger, use_hivemind_log_handler
 
+use_hivemind_log_handler("in_root_logger")
 logger = get_logger(__name__)