[WB-4663] Added Matplotlib guard (#1890)

wandb · Feb 26, 2021 · 181fc47 · 181fc47
1 parent eba2919
commit 181fc47
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/wandb/integration/fastai/__init__.py b/wandb/integration/fastai/__init__.py
@@ -44,7 +44,8 @@
 try:
     import matplotlib
 
-    matplotlib.use("Agg")  # non-interactive backend (avoid tkinter issues)
+    if wandb.wandb_lib.ipython._get_python_type() != "jupyter":
+        matplotlib.use("Agg")  # non-interactive backend (avoid tkinter issues)
     import matplotlib.pyplot as plt
 except:
     print("Warning: matplotlib required if logging sample image predictions")