Reinforcement Learning and Imitation Learning Environment Configuration

[TONGLoki]

Hello,

Currently I am trying to use imitation learning in surgical_robotics_challenge, is there a wrapped environment model for this project? I only find AmbfEnv() to make env in AMBF2.0.

from ambf_comm import AmbfEnv
ENV_NAME = 'psm1/baselink'# Training
env = AmbfEnv()
env.make(ENV_NAME)
env.reset()

Thank you!