[Enhancement] Refine import error of PAVI (#2098)

* Import Error message refined.

* PAVI import error message refined.
pull/2097/merge
Evelyn Wang 2022-07-05 14:23:19 +08:00 committed by GitHub
parent 9fffcbc9cd
commit c67ab9a356
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 1 deletions

View File

@ -68,7 +68,9 @@ class PaviLoggerHook(LoggerHook):
try:
from pavi import SummaryWriter
except ImportError:
raise ImportError('Please run "pip install pavi" to install pavi.')
raise ImportError(
'No module named pavi, please contact pavi team or visit'
'document for pavi installation instructions.')
self.run_name = runner.work_dir.split('/')[-1]