# -*- coding: utf-8 -*-
from models import *
from envs import *


class PerfectInfoAgent():
    def __init__(self, env, settings):
        super(PerfectInfoAgent, self).__init__()
        self.env = env
        self.solver = PerfectInfoOptimization(env)
        _, self.sol, _ = self.solver.solve()
        self.sol = self.sol.astype(int)

    def learn(self, epochs = 1000):
        pass

    def get_action(self, obs):
        return list(self.sol[:,self.env.current_step])