fixing matplotlib deprecation warning in plot() replicate#562

Signed-off-by: Enoch Kan <[email protected]>
enochkan · Mar 15, 2021 · b57c578 · b57c578
1 parent 9c69afa
commit b57c578
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/python/keepsake/experiment.py b/python/keepsake/experiment.py
@@ -442,8 +442,7 @@ def plot(self, metric: Optional[str] = None, logy=False):
 
         if metric is None:
             metric = self.primary_metric()
-
-        plotted_label = plt.axes().yaxis.get_label().get_text() or metric
+        plotted_label = plt.gca().get_ylabel() or metric
 
         if metric != plotted_label:
             plt.figure()