Robomimic environment wrapper (for Orbit)?

Hi there!
I wonder if there will be an environment wrapper for Robomimic library in Orbit? I defined a custom environment and I am trying to perform training of an agent on my collected data, and I can’t get to the training part, because I can’t create an environment. And even the tutorial example is not working for me. Did anyone succeeded in training on a custom environment?