Merge pull request #172 from liuzhenqi77/fix-plotting

[FIX] Fix plotting issue
netneurolab · Sep 18, 2024 · cca0cbb · cca0cbb
2 parents af78397 + 9f11acb
commit cca0cbb
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/neuromaps/nulls/__init__.py b/neuromaps/nulls/__init__.py
@@ -1,7 +1,7 @@
 """Functions for computing null models."""
 
 __all__ = [
-    'naive_nonparametric', 'alexander_bloch', 'vazquez_rodriguez', 'vasa',
+    'alexander_bloch', 'vazquez_rodriguez', 'vasa',
     'hungarian', 'baum', 'cornblath', 'burt2018', 'burt2020', 'moran'
 ]
 

diff --git a/neuromaps/plotting.py b/neuromaps/plotting.py
@@ -82,10 +82,13 @@ def plot_surf_template(data, template, density, surf='inflated',
             ax.disable_mouse_rotation()
             plot_surf(geom, img, hemi=HEMI[hemi], axes=ax, view=view, **opts)
             poly = ax.collections[0]
-            poly.set_facecolors(
-                _fix_facecolors(ax, poly._original_facecolor,
-                                *geom, view, hemi)
-            )
+            try:
+                poly.set_facecolors(
+                    _fix_facecolors(ax, poly._original_facecolor,
+                                    *geom, view, hemi)
+                )
+            except AttributeError:
+                pass
 
     if not opts.get('colorbar', False):
         fig.tight_layout()