diff --git a/pykat/finesse.py b/pykat/finesse.py index e2c8f51d518e80aafc66783254cf459c2159cb0f..71a97bb7b88beeef54a2a6a1d82f4d02f77bc215 100644 --- a/pykat/finesse.py +++ b/pykat/finesse.py @@ -71,6 +71,7 @@ class katRun(object): def __getitem__(self, value): idx = [i for i in range(len(self.ylabels)) if self.ylabels[i].split(" ")[0] == str(value)] + print idx if len(idx) == 1 and len(self.y.shape) == 1: return self.y.squeeze()