From a470994686033d2b804ef9c4a0a3f755f36509cb Mon Sep 17 00:00:00 2001
From: yh_cc <poemsmileyh@gmail.com>
Date: Tue, 10 May 2022 22:30:26 +0800
Subject: [PATCH] print bug fix

---
 fastNLP/core/log/print.py    | 2 +-
 tests/core/log/test_print.py | 8 ++++++++
 2 files changed, 9 insertions(+), 1 deletion(-)
 create mode 100644 tests/core/log/test_print.py

diff --git a/fastNLP/core/log/print.py b/fastNLP/core/log/print.py
index 42ac5062..827835aa 100644
--- a/fastNLP/core/log/print.py
+++ b/fastNLP/core/log/print.py
@@ -24,4 +24,4 @@ def print(*args, sep=' ', end='\n', file=None, flush=False):
     line = sep.join(map(str, args))
     if logger.isEnabledFor(INFO):
         kwargs = logger._add_rank_info({})
-        logger._log(INFO, line, **kwargs)
+        logger._log(INFO, line, None, **kwargs)
diff --git a/tests/core/log/test_print.py b/tests/core/log/test_print.py
new file mode 100644
index 00000000..8b5b3fa1
--- /dev/null
+++ b/tests/core/log/test_print.py
@@ -0,0 +1,8 @@
+from fastNLP import print
+
+
+def test_print():
+    print("a")
+    print([1, 2, 3])
+    print([1,2,3], [4,5,6], 'a')
+    print(print)