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)