Recently, we merged refactor!: environment interface is part of experiment code by tristanls-tbp · Pull Request #953 · thousandbrainsproject/tbp.monty · GitHub.
That pull request clarifies that the environment interface code in tbp.monty.frameworks.environments.embodied_data.py is experiment code that sets up the environment before experiment steps/episodes/epochs.
So, that code is moved to tbp.monty.experiment.environment.py.
tbp.monty.experiment is a new module, but I wanted to mark the place where we hope to eventually migrate all experiment-specific code so that it is easier for others to start moving things there.