Changeset 643 for branches

Timestamp:

01/13/13 17:31:32 (3 years ago)

Author:

mmckerns

Message:

enable discrete to take int or tuple for index

File:

• branches/decorate/wrapper.py (modified) (2 diffs)

branches/decorate/wrapper.py

@@ -131 +131 @@
 [1.0, 2.0, 2.0]
 
->>> @discrete([1,3,5,7], index=(0,3)])
+>>> @discrete([1,3,5,7], index=(0,3))
 ... def squared(x):
 ....    return [i**2 for i in x]
@@ -139 +139 @@
     samples = [asarray(samples)]
     samples[0].sort()
+    if isinstance(index, int): index = (index,)
     index = [index]