Создание среды для обучения агента в Python

Здраствуйте, прошу подсказать, как и где лучше создать среду, для обучения агента(ов) в рамках обучения с подкреплением. У меня есть среда в pygame, в которой агент двигается перемещается к заданной точке обходя динамические препятствия, можно ли эту среду адаптировать под RL, или как это удобно сделать(можно ли создать уникальную среду например в gym)