From d5c8bafef70d0e83093befe5e6505a32e6f8eaee Mon Sep 17 00:00:00 2001
From: chenxuyi <chenxuyi@baidu.com>
Date: Fri, 21 May 2021 20:07:47 +0800
Subject: [PATCH] +logging format

---
 ernie/__init__.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/ernie/__init__.py b/ernie/__init__.py
index 9171890..f98ba81 100644
--- a/ernie/__init__.py
+++ b/ernie/__init__.py
@@ -17,6 +17,9 @@ from __future__ import absolute_import
 from __future__ import print_function
 from __future__ import unicode_literals
 
+import sys
+import logging
+
 import paddle
 if paddle.__version__ != '0.0.0' and paddle.__version__ < '2.0.0':
     raise RuntimeError('propeller 0.2 requires paddle 2.0+, got %s' %
@@ -28,3 +31,10 @@ from ernie.modeling_ernie import (
     ErnieModelForQuestionAnswering, ErnieModelForPretraining)
 
 from ernie.tokenizing_ernie import ErnieTokenizer, ErnieTinyTokenizer
+
+log = logging.getLogger(__name__)
+formatter = logging.Formatter(fmt='[%(levelname)s] %(asctime)s [%(filename)12s:%(lineno)5d]:    %(message)s')
+stream_hdl = logging.StreamHandler(stream=sys.stderr)
+stream_hdl.setFormatter(formatter)
+log.addHandler(stream_hdl)
+log.propagate = False
-- 
GitLab