fix matplotlib warning

iterative · Sep 7, 2023 · 6ccc959 · 6ccc959
1 parent a02d160
commit 6ccc959
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/dvclive/live.py b/src/dvclive/live.py
@@ -247,7 +247,7 @@ def _init_report(self):
                     " inside a notebook. Disabling report."
                 )
                 self._report_mode = None
-        if self._report_mode != "html" and not matplotlib_installed():
+        if self._report_mode in ("notebook", "md") and not matplotlib_installed():
             logger.warning(
                 f"Report mode '{self._report_mode}' requires 'matplotlib'"
                 " to be installed. Disabling report."