Провести тренировку RL-моделей

Есть сформированные 2 модели: EpsilonGreedyAlgorithm и MonteCarloTreeSearch для игры https://ru.wikipedia.org/wiki/%D0%A7%D0%B5%D1%82%D...
Необходимо адаптировать модели под новые данные в готовом коде с GitHub и получить рабочую версию.
Есть тестовые данные. Код написан с использованием библиотек OpenAI Gym, NumPy, PyTorch, matplotlib. В коде есть доходчивый README.md, отправлю документацию к каждой из библиотек.