Implement the ML-agent package within a Tower Defence Unity game

I currently have a ready-to-play Tower defence game on the Unity platform and I was looking to implement reinforcement learning in the game using the ML-agent package. I am looking to implement the package on just the first level of the game, not the full game.