uniswap-model.py

class UniswapModel(Model):
    def __init__(self, num_traders, num_arbitrageurs, trader_dai, trader_eth, uniswap_dai, uniswap_eth):
        super().__init__()
        self.num_traders = num_traders
        self.schedule = RandomActivation(self)
        self.uniswap = Uniswap(1, self, uniswap_dai, uniswap_eth)

        for i in range(num_traders):
            trader = Trader(i, self, trader_eth, trader_dai, i < num_arbitrageurs)
            self.schedule.add(trader)

        self.datacollector = DataCollector(model_reporters=model_reporters,agent_reporters=agent_reporters)

    def step(self):
        self.datacollector.collect(self)
        self.uniswap.step()
        self.schedule.step()