steffen/marl-factory-grid
1
0
Fork 0
mirror of https://github.com/illiumst/marl-factory-grid.git synced 2025-12-14 03:00:37 +01:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Julian Schönberger
2024-08-09 16:36:05 +02:00
parent 973f3e9fc8
commit 0a8eaf8827
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
marl_factory_grid/algorithms/tsp/TSP_base_agent.py
View File

@@ -90,7 +90,7 @@ class TSPBaseAgent(ABC):
[self.state.pos] + \ [self.state.pos] + \
[x for x in positions if max(abs(np.subtract(x, self.state.pos))) < 3] [x for x in positions if max(abs(np.subtract(x, self.state.pos))) < 3]
try: try:
while len(nodes) < 7: while len(nodes) < 13:
nodes += [next(x for x in positions if x not in nodes)] nodes += [next(x for x in positions if x not in nodes)]
except StopIteration: except StopIteration:
nodes = [self.state.pos] + positions nodes = [self.state.pos] + positions