Fixed scoping error with Transitions
This commit is contained in:
parent
ad75539776
commit
07c90a09f9
1 changed files with 5 additions and 1 deletions
|
@ -1,4 +1,8 @@
|
|||
from .PrioritizedReplayMemory import PrioritizedReplayMemory, Transition
|
||||
from .PrioritizedReplayMemory import PrioritizedReplayMemory
|
||||
|
||||
Transition = namedtuple('Transition',
|
||||
('state', 'action', 'reward', 'next_state', 'done'))
|
||||
|
||||
|
||||
class DQfDMemory(PrioritizedReplayMemory):
|
||||
def __init__(self, capacity, alpha):
|
||||
|
|
Loading…
Reference in a new issue