diff --git a/agent.py b/agent.py index 86ee113..0f174a9 100644 --- a/agent.py +++ b/agent.py @@ -117,13 +117,6 @@ class Agent: return else: next = self.path.pop() - if (next.x, next.y) == dest_coords: - if self.dest.package: - self.pick_up_package(self.dest.package) - return - elif self.dest.is_rack: - self.unload_package(self.dest) - return star_dir = self.direction if self.x > next.x and not self.direction == 'left': if self.direction == 'down': @@ -145,7 +138,15 @@ class Agent: self.turn_right() else: self.turn_left() - + + if (next.x, next.y) == dest_coords: + if self.dest.package: + self.pick_up_package(self.dest.package) + return + elif self.dest.is_rack: + self.unload_package(self.dest) + return + if star_dir == self.direction: self.x = next.x self.y = next.y