Browse Source

print bug fix

tags/v1.0.0alpha
yh_cc 2 years ago
parent
commit
a470994686
2 changed files with 9 additions and 1 deletions
  1. +1
    -1
      fastNLP/core/log/print.py
  2. +8
    -0
      tests/core/log/test_print.py

+ 1
- 1
fastNLP/core/log/print.py View File

@@ -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)

+ 8
- 0
tests/core/log/test_print.py View File

@@ -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)

Loading…
Cancel
Save