mirror of
https://github.com/illiumst/marl-factory-grid.git
synced 2025-05-22 14:56:43 +02:00
16 lines
385 B
Python
16 lines
385 B
Python
from random import randint
|
|
|
|
from mfg_package.algorithms.static.TSP_base_agent import TSPBaseAgent
|
|
|
|
future_planning = 7
|
|
|
|
|
|
class TSPRandomAgent(TSPBaseAgent):
|
|
|
|
def __init__(self, n_actions, *args, **kwargs):
|
|
super(TSPRandomAgent, self).__init__(*args, **kwargs)
|
|
self.n_action = n_actions
|
|
|
|
def predict(self, *_, **__):
|
|
return randint(0, self.n_action - 1)
|