diff options
| -rw-r--r-- | boltzgen/geometry.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/boltzgen/geometry.py b/boltzgen/geometry.py index 7cc406d..d4411ad 100644 --- a/boltzgen/geometry.py +++ b/boltzgen/geometry.py @@ -24,8 +24,8 @@ class Geometry: else: raise Exception('Geometry malformed') - def inner_cells(self): - for idx in numpy.ndindex(self.inner_size()): + def inner_cells(self, order='F'): + for idx in ndindex(self.inner_size(), order): yield tuple(map(lambda i: i + 1, idx)) def size(self): |