Commit 4ac62423 authored by Daniel Brown's avatar Daniel Brown
Browse files

fixing output selector

parent 50d0ed53
......@@ -75,14 +75,14 @@ class katRun(object):
def __getitem__(self, value):
idx = [i for i in range(len(self.ylabels)) if self.ylabels[i].split(" ")[0] == str(value)]
if len(idx) > 0 and self.y.shape == ():
if len(idx) > 0 and self.y.shape == ():
# In the case we have a noxaxis and just one output...
return float(self.y)
elif len(idx) == 1 and len(self.y.shape) >= 1:
elif len(idx) == 1 and len(self.y.shape) == 1:
# only selecting a single output from a 1D array
if self.y.size == 1:
if self.y.size == 1:
return self.y
else:
return self.y[idx[0]]
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment