Merge pull request #1253 from enthought/fix_non_integer_shape

MAINT: fix non integer shape
enthought · May 23, 2024 · c95f140 · c95f140
2 parents 284c8e9 + e6f3153
commit c95f140
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/mayavi/mlab/simple_structured_grid.py b/examples/mayavi/mlab/simple_structured_grid.py
@@ -43,10 +43,10 @@
 # We reorder the points, scalars and vectors so this is as per VTK's
 # requirement of x first, y next and z last.
 pts = pts.transpose(2, 1, 0, 3).copy()
-pts.shape = pts.size / 3, 3
+pts.shape = pts.size // 3, 3
 scalars = scalars.T.copy()
 vectors = vectors.transpose(2, 1, 0, 3).copy()
-vectors.shape = vectors.size / 3, 3
+vectors.shape = vectors.size // 3, 3
 
 # Create the dataset.
 sg = tvtk.StructuredGrid(dimensions=x.shape, points=pts)