Fixed a minor bug

polecart/main/agent.py

@@ -374,12 +374,12 @@ class Agent:
 					)
 				)
 
+				state = next_state
 
 				# Only train the network if we have enough
 				# transitions in memory to do so.
 				if len(self.memory) >= self.BATCH_SIZE:
 
-					state = next_state
 					
 					# Run optimizer
 					self._optimize()