From 32102b1ea6a240137f73dc7bca778cfe4bc52958 Mon Sep 17 00:00:00 2001
From: Eric Larson <larson.eric.d@gmail.com>
Date: Sat, 10 Jun 2023 08:01:32 -0400
Subject: [PATCH] MAINT: Update for PyVista deprecation

---
 mne/viz/backends/_pyvista.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/mne/viz/backends/_pyvista.py b/mne/viz/backends/_pyvista.py
index 05d2aa132e7..affb1f34bfd 100644
--- a/mne/viz/backends/_pyvista.py
+++ b/mne/viz/backends/_pyvista.py
@@ -993,7 +993,10 @@ def _volume(
         center,
     ):
         # Now we can actually construct the visualization
-        grid = pyvista.UniformGrid()
+        try:
+            grid = pyvista.ImageData()
+        except AttributeError:  # PV < 0.40
+            grid = pyvista.UniformGrid()
         grid.dimensions = dimensions + 1  # inject data on the cells
         grid.origin = origin
         grid.spacing = spacing