Skip to content
Merged
Prev Previous commit
removed redundant code, fixed a comment
  • Loading branch information
bakerwho committed Feb 28, 2018
commit 1a1a0495073d463f2c5bae0e45587d0151ea9dee
5 changes: 2 additions & 3 deletions rl.py
Original file line number Diff line number Diff line change
Expand Up @@ -35,16 +35,15 @@ def __init__(self, pi, mdp):
self.Ns1_sa = defaultdict(int)
self.s = None
self.a = None
self.visited = set()
self.visited = set() # keeping track of visited states

def __call__(self, percept):
s1, r1 = percept
self.mdp.states.add(s1) # Model keeps track of visited states.
mdp = self.mdp
R, P, terminals, pi = mdp.reward, mdp.P, mdp.terminals, self.pi
s, a, Nsa, Ns1_sa, U = self.s, self.a, self.Nsa, self.Ns1_sa, self.U

if s1 not in self.visited: # Reward is only available for visited state.
if s1 not in self.visited: # Reward is only known for visited state.
U[s1] = R[s1] = r1
self.visited.add(s1)
if s is not None:
Expand Down