diff --git a/docker_train.sh b/docker_train.sh index a7ce60f..1dae1a7 100644 --- a/docker_train.sh +++ b/docker_train.sh @@ -9,7 +9,7 @@ if [ "$1" == "lunarlander" ]; then python run_lunarlander_continuous.py --algo $2 --off-render elif [ "$1" == "openmanipulator" ]; then cd $KAIR/scripts; \ - rosrun kair_algorithms run_open_manipulator_reacher_v0.py --algo $2 --off-render + /opt/ros/$ROS_DISTRO/bin/rosrun kair_algorithms run_open_manipulator_reacher_v0.py --algo $2 --off-render else echo "Unknown parameter" fi