Skip to content

Commit a576534

Browse files
ezeeetmvrkhare
authored andcommitted
fixes aws#775 (aws#777)
* fixes aws#775 aws#775 * updated TSP_env.py aws#777 (review)
1 parent af59c4c commit a576534

File tree

1 file changed

+6
-1
lines changed
  • reinforcement_learning/rl_traveling_salesman_vehicle_routing_coach/src

1 file changed

+6
-1
lines changed

reinforcement_learning/rl_traveling_salesman_vehicle_routing_coach/src/TSP_env.py

+6-1
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,12 @@ def reset(self):
9696
self.agt_x = self.restaurant_x
9797
self.agt_y = self.restaurant_y
9898
if self.randomized_orders:
99-
orders = [self.__receive_order() for i in range(self.n_orders)]
99+
# Enforce uniqueness of orders, to prevent multiple orders being placed on the same points
100+
# And ensure actual orders in the episode are always == n_orders as expected
101+
orders=[]
102+
while len(orders) != self.n_orders:
103+
orders += [self.__receive_order()]
104+
orders = list(set(orders))
100105
else:
101106
orders = [(-2, -2), (1,1), (2,0), (0, -2)]
102107
self.o_x = [x for x, y in orders]

0 commit comments

Comments
 (0)