fix: make indexing compatible with py3.10

inducer · Oct 12, 2024 · 84d5e08 · 84d5e08
1 parent 6bfbc81
commit 84d5e08
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/meshmode/discretization/poly_element.py b/meshmode/discretization/poly_element.py
@@ -536,8 +536,9 @@ def quadrature_rule(self):
             nodes_tp = self._nodes
 
         for idim, (nodes, basis) in enumerate(zip(nodes_tp, self._basis.bases)):
-            # get current dimension's nodes from fastest varying axis
-            nodes = nodes[*(0,)*idim, :, *(0,)*(self.dim-idim-1)]
+            # get current dimension's nodes
+            iaxis = (*(0,)*idim, slice(None), *(0,)*(self.dim-idim-1))
+            nodes = nodes[iaxis]
 
             nodes_1d = nodes.reshape(1, -1)
             mass_matrix = mp.mass_matrix(basis, nodes_1d)