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

* Import Error message refined. * PAVI import error message refined.

[Enhancement] Refine import error of PAVI (#2098)
* Import Error message refined. * PAVI import error message refined.
c67ab9a3 · Evelyn Wang · GitHub · 9fffcbc9 · c67ab9a3
Unverified Commit c67ab9a3 authored Jul 05, 2022 by Evelyn Wang Committed by GitHub Jul 05, 2022
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

mmcv/runner/hooks/logger/pavi.py mmcv/runner/hooks/logger/pavi.py +3 -1

No files found.
--- a/mmcv/runner/hooks/logger/pavi.py
+++ b/mmcv/runner/hooks/logger/pavi.py
@@ -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]