snapshot...

modules/util/plot.py

@@ -330,7 +330,7 @@ def make_time_domain_hist(values, edges):
     plt.show()
 
 
-def scatter_density(x, y, color=None, fname=None, fig=None, x_rng=None, y_rng=None, dpi=72):
+def scatter_density(x, y, color=None, fname=None, fig=None, ax=None, x_rng=None, y_rng=None, dpi=72):
     if fig is None:
         fig = plt.figure()
         ax = fig.add_subplot(1, 1, 1, projection='scatter_density')
@@ -345,4 +345,4 @@ def scatter_density(x, y, color=None, fname=None, fig=None, x_rng=None, y_rng=No
     if fname is not None:
         fig.savefig(fname)
 
-    return fig
+    return fig, ax
\ No newline at end of file