From 89416c6ce888d73a773d6bff8fe65695719c703b Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Tue, 5 Jan 2016 18:58:17 -0600
Subject: [PATCH] indices_in_shape: Accept pure-integer shapes
---
pytools/__init__.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/pytools/__init__.py b/pytools/__init__.py
index c744b32..ed02ed0 100644
--- a/pytools/__init__.py
+++ b/pytools/__init__.py
@@ -1072,6 +1072,9 @@ def wandering_element(length, wanderer=1, landscape=0):
def indices_in_shape(shape):
+ if isinstance(shape, int):
+ shape = (shape,)
+
if len(shape) == 0:
yield ()
elif len(shape) == 1:
--
GitLab