Fix dr-logs-robomaker for compse style

aws-deepracer-community · Jan 7, 2024 · e49d74d · e49d74d
1 parent 22efcad
commit e49d74d
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/bin/scripts_wrapper.sh b/bin/scripts_wrapper.sh
@@ -243,7 +243,13 @@ function dr-find-robomaker {
   esac
   done
 
-  eval ROBOMAKER_ID=$(docker ps | grep "${OPT_PREFIX}-${DR_RUN_ID}_robomaker.${OPT_REPLICA}" | cut -f1 -d\  | head -1)
+  if [[ "${DR_DOCKER_STYLE,,}" == "swarm" ]];
+  then
+    eval ROBOMAKER_ID=$(docker ps | grep "${OPT_PREFIX}-${DR_RUN_ID}_robomaker.${OPT_REPLICA}" | cut -f1 -d\  | head -1)
+  else
+    eval ROBOMAKER_ID=$(docker ps | grep "${OPT_PREFIX}-${DR_RUN_ID}-robomaker-${OPT_REPLICA}" | cut -f1 -d\  | head -1)  
+  fi
+
   if [ -n "$ROBOMAKER_ID" ]; then
     echo $ROBOMAKER_ID
   fi