dataset generation for custom environment

[yorgoon]

Hello,

I am interested in using offline RL for medical robotics, specifically using dVRK. In order to be compatible with the gym, I made a custom environment to apply RL algorithms. What are the necessary methods to be implemented in the custom gym environment so as to utilize dataset generation functionality in your package?

Best, Soowon