Entity Gym is an open source Python library that defines an entity-based API for reinforcement learning environments. Entity Gym extends the standard paradigm of fixed-size observation spaces by allowing observations to contain dynamically-sized lists of entities.
If we can create a wrapper that implements the EntityGym API, then it may help other researchers use our environment. Not sure if this wrapper should be implemented on the environment or in the agents, though 🤔