Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def __init__(self, name, team, index):
from leviathan.output_formatter import OutputFormatter
from leviathan.input_formatter import InputFormatter
from leviathan.cool_atba import Atba
import torch
BaseAgent.__init__(self, name, team, index)
self.pipe = None
self.actor_model = None
self.team_model = None
self.game_memory = None
self.atba = Atba()
self.torch = torch
self.output_formatter = OutputFormatter()
self.input_formatter = InputFormatter(self.index, self.index)
# self.input_formatter = InputFormatter(self.index, (self.index + 1) % 2)