Added Matplotlib guard
@@ -44,6 +44,7 @@ import random
try:
import matplotlib
+ if wandb.wandb_lib.ipython._get_python_type() != "jupyter":
matplotlib.use("Agg") # non-interactive backend (avoid tkinter issues)
import matplotlib.pyplot as plt
except:
