In [19]:

    

from transition_table import TransitionTable
from action import Action
import numpy as np


    

In [20]:

    

table = TransitionTable()


    

In [21]:

    

img = np.random.rand(84, 84)
table.add_transition(img, False, Action(2), 0.5, True)


    

    




---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-21-0b16171f96c0> in <module>()
      1 img = np.random.rand(84, 84)
----> 2 table.add_transition(img, False, Action(2), 0.5, True)

/Users/raphie/Dropbox/Code/cs231n-project/transition_table.py in add_transition(self, *args)
     63                 return s, t, a, r, sp
     64 
---> 65         def add_transition(self, *args):
     66                 t = Transition(*args)
     67                 self.transitions.append(t)

/Users/raphie/Dropbox/Code/cs231n-project/transition_table.py in __init__(self, action, reward, image, terminal, was_start)
      6 
      7 class Transition(object):
----> 8         def __init__(self, image, terminal, action, reward, was_start):
      9                 pdb.set_trace()
     10                 self.action = action

ValueError: too many values to unpack

In [ ]: