Fixes for matplotlib 3.9 API (#8)

* Fix "x must be a sequence" * Fix for matplotlib 3.9.0 api change
xsuite · Jun 3, 2024 · 5dfdd81 · 5dfdd81
1 parent 8711b55
commit 5dfdd81
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/xplt/hooks.py b/xplt/hooks.py
@@ -35,10 +35,10 @@ def register_matplotlib_options():
         cmap_petroff_bipolar,
         cmap_petroff_cyclic,
     ):
-        mpl.cm.register_cmap(cmap=cmap)
+        mpl.colormaps.register(cmap=cmap)
         cmap_r = cmap.reversed()
         cmap_r.name = cmap.name + "_r"
-        mpl.cm.register_cmap(cmap=cmap_r)
+        mpl.colormaps.register(cmap=cmap_r)
 
 
 def register_pint_options():

diff --git a/xplt/phasespace.py b/xplt/phasespace.py
@@ -383,7 +383,7 @@ def update(self, particles, *, mask=None, masks=None, autoscale=False):
 
             # 2D mean indicator (cross)
             if self.artists_mean[i]:
-                self.artists_mean[i].set_data(XY0)
+                self.artists_mean[i].set_data(XY0.reshape([2,1]))
                 changed_artists.append(self.artists_mean[i])
 
             # 2D size indicator (ellipses)