Merge pull request #4063 from autowarefoundation/main

merge main branch changes
autowarefoundation · Dec 14, 2023 · c71e075 · c71e075
2 parents c9c4546 + d4dccbc
commit c71e075
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/setup-dev-env.sh b/setup-dev-env.sh
@@ -38,6 +38,10 @@ while [ "$1" != "" ]; do
         option_data_dir="$2"
         shift
         ;;
+    --download-artifacts)
+        # Set download artifacts option
+        option_download_artifacts=true
+        ;;
     *)
         args+=("$1")
         ;;
@@ -85,7 +89,7 @@ fi
 
 # Check installation of CUDA Drivers
 if [ "$option_no_cuda_drivers" = "true" ]; then
-    ansible_args+=("--extra-vars" "install_cuda_drivers=false")
+    ansible_args+=("--extra-vars" "cuda_install_drivers=false")
 fi
 
 # Check installation of dev package
@@ -97,6 +101,12 @@ else
     ansible_args+=("--extra-vars" "tensorrt_install_devel=true")
 fi
 
+# Check downloading artifacts
+if [ "$option_yes" = "true" ] || [ "$option_download_artifacts" = "true" ]; then
+    echo -e "\e[36mArtifacts will be downloaded to $option_data_dir\e[m"
+    ansible_args+=("--extra-vars" "prompt_download_artifacts=y")
+fi
+
 ansible_args+=("--extra-vars" "data_dir=$option_data_dir")
 
 # Load env