[GH-16333] fix pyplot warning (#16381) (#16441)

Co-authored-by: krasinski <[email protected]>
h2oai · Oct 30, 2024 · 421def8 · 421def8
1 parent 4056d47
commit 421def8
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/h2o-py/h2o/plot/_matplotlib.py b/h2o-py/h2o/plot/_matplotlib.py
@@ -5,7 +5,10 @@ def get_matplotlib_pyplot(server, raise_if_not_available=False):
         # noinspection PyUnresolvedReferences
         import matplotlib
         if server:
-            matplotlib.use("Agg")
+            if matplotlib.get_backend() != "Agg":
+                import matplotlib.pyplot as plt
+                plt.close('all')
+                matplotlib.use("Agg")
         try:
             # noinspection PyUnresolvedReferences
             import matplotlib.pyplot as plt