Bug fix AppliedDataSciencePartners#3

Shatterblade · Oct 9, 2018 · 91e6605 · 91e6605
1 parent 1a4ba4a
commit 91e6605
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 15 deletions.
diff --git a/.idea/workspace.xml b/.idea/workspace.xml
diff --git a/agent.py b/agent.py
@@ -27,10 +27,11 @@ def act(self, state, tau):
 		state.print()
 		time.sleep(2)
 		action = -1
+		allowedActions = state.allowedActions
 		while 1:
 			try:
 				action = int(input('Enter your move: '))
-				if state.allowedActions.__contains__(action):
+				if allowedActions.__contains__(action):
 					break
 				else:
 					print('Enter a valid move.')