<s> __author__ = '<STR_LIT>' <EOL> from scipy import zeros , array <EOL> from . maze import MazeTask <EOL> class CheeseMaze ( MazeTask ) : <EOL> """<STR_LIT>""" <EOL> observations = <NUM_LIT:7> <EOL> discount = <NUM_LIT> <EOL> topology = array ( [ [ <NUM_LIT:1> ] * <NUM_LIT:7> , <EOL> [ <NUM_LIT:1> , <NUM_LIT:0> , <NUM_LIT:1> , <NUM_LIT:0> , <NUM_LIT:1> , <NUM_LIT:0> , <NUM_LIT:1> ] , <EOL> [ <NUM_LIT:1> , <NUM_LIT:0> , <NUM_LIT:1> , <NUM_LIT:0> , <NUM_LIT:1> , <NUM_LIT:0> , <NUM_LIT:1> ] , <EOL> [ <NUM_LIT:1> , <NUM_LIT:0> , <NUM_LIT:0> , <NUM_LIT:0> , <NUM_LIT:0> , <NUM_LIT:0> , <NUM_LIT:1> ] , <EOL> [ <NUM_LIT:1> ] * <NUM_LIT:7> ] ) <EOL> goal = ( <NUM_LIT:1> , <NUM_LIT:3> ) <EOL> def getObservation ( self ) : <EOL> """<STR_LIT>""" <EOL> res = zeros ( <NUM_LIT:7> ) <EOL> obs = self . env . getSensors ( ) <EOL> if self . env . perseus == self . env . goal : <EOL> res [ <NUM_LIT:6> ] = <NUM_LIT:1> <EOL> elif sum ( obs ) == <NUM_LIT:3> : <EOL> res [ <NUM_LIT:0> ] = <NUM_LIT:1> <EOL> elif sum ( obs ) == <NUM_LIT:1> : <EOL> res [ <NUM_LIT:5> ] = <NUM_LIT:1> <EOL> elif obs [ <NUM_LIT:0> ] == obs [ <NUM_LIT:1> ] : <EOL> res [ <NUM_LIT:1> ] = <NUM_LIT:1> <EOL> elif obs [ <NUM_LIT:0> ] == obs [ <NUM_LIT:3> ] : <EOL> res [ <NUM_LIT:2> ] = <NUM_LIT:1> <EOL> elif obs [ <NUM_LIT:0> ] == obs [ <NUM_LIT:2> ] : <EOL> if obs [ <NUM_LIT:0> ] == <NUM_LIT:1> : <EOL> res [ <NUM_LIT:3> ] = <NUM_LIT:1> <EOL> else : <EOL> res [ <NUM_LIT:4> ] = <NUM_LIT:1> <EOL> return res </s>
