From 07c90a09f94107a7dba7564554ad8ab0c205d8e2 Mon Sep 17 00:00:00 2001 From: Brandon Rozek Date: Mon, 4 Nov 2019 12:09:09 -0500 Subject: [PATCH] Fixed scoping error with Transitions --- rltorch/memory/DQfDMemory.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/rltorch/memory/DQfDMemory.py b/rltorch/memory/DQfDMemory.py index 73ce175..ba957ed 100644 --- a/rltorch/memory/DQfDMemory.py +++ b/rltorch/memory/DQfDMemory.py @@ -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):