From 5131f6b0a60a11980236edda2bfa0ba29c8bcb5a Mon Sep 17 00:00:00 2001
From: Varun Agrawal <varagrawal@gmail.com>
Date: Fri, 3 Sep 2021 08:05:11 -0400
Subject: [PATCH] fix matplotlib deprecation

---
 python/gtsam/utils/plot.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/python/gtsam/utils/plot.py b/python/gtsam/utils/plot.py
index f324da63a..7ea393077 100644
--- a/python/gtsam/utils/plot.py
+++ b/python/gtsam/utils/plot.py
@@ -23,7 +23,10 @@ def set_axes_equal(fignum: int) -> None:
       fignum: An integer representing the figure number for Matplotlib.
     """
     fig = plt.figure(fignum)
-    ax = fig.gca(projection='3d')
+    if not fig.axes:
+        ax = fig.add_subplot(projection='3d')
+    else:
+        ax = fig.axes[0]
 
     limits = np.array([
         ax.get_xlim3d(),
@@ -339,7 +342,11 @@ def plot_pose3(
     """
     # get figure object
     fig = plt.figure(fignum)
-    axes = fig.gca(projection='3d')
+    if not fig.axes:
+        axes = fig.add_subplot(projection='3d')
+    else:
+        axes = fig.axes[0]
+
     plot_pose3_on_axes(axes, pose, P=P, axis_length=axis_length)
 
     axes.set_xlabel(axis_labels[0])