I was following this tutorial on the ML-Agents package for Unity: https://github.com/Unity-Technologies/ml-agents/blob/release_12_docs/docs/Learning-Environment-Create-New.m