added default rasterization to scatter_plot_2d()

handley-lab · Apr 17, 2024 · 3892b52 · 3892b52
1 parent 79c7bb6
commit 3892b52
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/anesthetic/plot.py b/anesthetic/plot.py
@@ -1431,6 +1431,10 @@ def scatter_plot_2d(ax, data_x, data_y, *args, **kwargs):
           effective samples in :func:`anesthetic.utils.neff`
           with ``beta=ncompress``.
 
+    Other Parameters
+    ----------------
+    **kwargs : rasterized = True (by default we rasterize scatter plots)
+
     Returns
     -------
     lines : :class:`matplotlib.lines.Line2D`
@@ -1454,10 +1458,10 @@ def scatter_plot_2d(ax, data_x, data_y, *args, **kwargs):
     cmap = kwargs.pop('cmap', None)
     color = kwargs.pop('color', (ax._get_lines.get_next_color()
                                  if cmap is None else cmap(0.68)))
-
+    rasterized = kwargs.pop('raster', True) 
     kwargs.pop('q', None)
 
-    points = ax.plot(data_x, data_y, 'o', color=color, markersize=markersize,
+    points = ax.plot(data_x, data_y, 'o', color=color, markersize=markersize, rasterized=rasterized,
                      *args, **kwargs)
     return points