def policy(w): ''' Calculate policy for given world''' p = Policy(w) p.policyIteration(turbo=True) return p