Fix deprecated gca argument

traja-team · Apr 6, 2024 · aea1b0a · aea1b0a
1 parent 963b4b8
commit aea1b0a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/traja/plotting.py b/traja/plotting.py
@@ -814,7 +814,7 @@ def plot_surface(
     Z = np.sqrt(U * U + V * V)
 
     fig = plt.figure()
-    ax = fig.gca(projection="3d")
+    ax = fig.add_subplot(projection='3d')
     ax.plot_surface(
         X, Y, Z, cmap= cmap, linewidth=0, **surfaceplot_kws
     )