Use create_probe_polygon for probeinterface v>=0.2.27

Author: alejoe91

src/spikeinterface/widgets/unit_waveforms.py

@@ -1,7 +1,8 @@
 from __future__ import annotations
 
-import numpy as np
+from packaging.version import parse
 from warnings import warn
+import numpy as np
 
 from .base import BaseWidget, to_attr
 from .utils import get_unit_colors
@@ -383,11 +384,16 @@ def plot_matplotlib(self, data_plot, **backend_kwargs):
 
             # plot channels
             if dp.plot_channels:
-                from probeinterface.plotting import create_probe_polygons
+                from probeinterface import __version__ as pi_version
 
-                probe = dp.sorting_analyzer_or_templates.get_probe()
-                contacts, probe_outline = create_probe_polygons(probe, contacts_colors="w")
-                ax.add_collection(contacts)
+                if parse(pi_version) >= parse("0.2.27"):
+                    from probeinterface.plotting import create_probe_polygons
+
+                    probe = dp.sorting_analyzer_or_templates.get_probe()
+                    contacts, _ = create_probe_polygons(probe, contacts_colors="w")
+                    ax.add_collection(contacts)
+                else:
+                    ax.scatter(dp.channel_locations[:, 0], dp.channel_locations[:, 1], color="k")
 
             # Apply axis_equal setting
             if dp.axis_equal: