From 3794673c704b18756ed28c836d7bab4be24c4f35 Mon Sep 17 00:00:00 2001 From: Ethan Peterson Date: Sun, 27 Nov 2022 15:54:03 -0500 Subject: [PATCH] Apply suggestions from code review Co-authored-by: Patrick Shriwise --- openmc/mesh.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/openmc/mesh.py b/openmc/mesh.py index d180bc7bd7..157e782633 100644 --- a/openmc/mesh.py +++ b/openmc/mesh.py @@ -172,7 +172,8 @@ class StructuredMesh(MeshBase): ------- vertices : numpy.ndarray Returns a numpy.ndarray representing the coordinates of the mesh - vertices with a shape equal to (ndim, dim1 + 1, ..., dimn + 1). + vertices with a shape equal to (ndim, dim1 + 1, ..., dimn + 1). Can be + unpacked along the first dimension with xx, yy, zz = mesh.vertices. """ return np.stack(np.meshgrid(*self._grids, indexing='ij'), axis=0) @@ -186,7 +187,8 @@ class StructuredMesh(MeshBase): centroids : numpy.ndarray Returns a numpy.ndarray representing the mesh element centroid coordinates with a shape equal to (ndim, dim1, ..., dimn). Can be - unpacked by the first dimension with xx, yy, zz = mesh.centroids + unpacked along the first dimension with xx, yy, zz = mesh.centroids. + """ ndim = self.n_dimension