Fix small bug in Ligating Loop when setting ObservationType to RGBD

ScheiklP / sofa_env

Reinforcement learning environments for robot-assisted laparoscopic surgery

MIT License

47 stars 14 forks source link

Fix small bug in Ligating Loop when setting ObservationType to RGBD #17

Closed erikbwu closed 9 months ago

erikbwu commented 9 months ago

To reproduce run the following:

from sofa_env.base import RenderMode
from sofa_env.scenes.ligating_loop.ligating_loop_env import LigatingLoopEnv, ObservationType, ActionType

env = LigatingLoopEnv(
    observation_type=ObservationType.RGBD,
    render_mode=RenderMode.HUMAN,
    action_type=ActionType.CONTINUOUS,
    image_shape=(256, 256),
    frame_skip=1,
    time_step=0.1,
    settle_steps=50,
)

ScheiklP commented 9 months ago

Hi @KlRE, thanks a lot for the PR! Good catch! :)

Cheers, Paul