Increased maximum number of nodes for TSP agent to deal with all potential dirt placements in dirt_quadrant

2025-07-08 02:21:36 +02:00 · 2024-08-09 16:36:05 +02:00
parent 973f3e9fc8
commit 0a8eaf8827
1 changed files with 1 additions and 1 deletions
--- a/marl_factory_grid/algorithms/tsp/TSP_base_agent.py
+++ b/marl_factory_grid/algorithms/tsp/TSP_base_agent.py
@ -90,7 +90,7 @@ class TSPBaseAgent(ABC):
                    [self.state.pos] + \
                    [x for x in positions if max(abs(np.subtract(x, self.state.pos))) < 3]
                try:
-                    while len(nodes) < 7:
+                    while len(nodes) < 13:
                        nodes += [next(x for x in positions if x not in nodes)]
                except StopIteration:
                    nodes = [self.state.pos] + positions