diff --git a/docker_train.sh b/docker_train.sh index 913b3d2..a7ce60f 100644 --- a/docker_train.sh +++ b/docker_train.sh @@ -6,9 +6,10 @@ KAIR=$CATKIN_WS/src/kair_algorithms_draft if [ "$1" == "lunarlander" ]; then cd $KAIR/scripts; \ - python run_lunarlander_continuous.py --algo $2 --off-render + python run_lunarlander_continuous.py --algo $2 --off-render elif [ "$1" == "openmanipulator" ]; then - echo "Working" + cd $KAIR/scripts; \ + rosrun kair_algorithms run_open_manipulator_reacher_v0.py --algo $2 --off-render else echo "Unknown parameter" fi