Update run-sparkrapids-xgboost.sh

NVIDIA · Nov 19, 2024 · 88ccbb6 · 88ccbb6
1 parent 60f8c9b
commit 88ccbb6
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/examples/ML+DL-Examples/Optuna-Spark/optuna-examples/standalone/run-sparkrapids-xgboost.sh b/examples/ML+DL-Examples/Optuna-Spark/optuna-examples/standalone/run-sparkrapids-xgboost.sh
@@ -1,5 +1,10 @@
 # Copyright (c) 2024, NVIDIA CORPORATION.
 
+if [[ -z $SPARK_HOME ]]; then
+  echo "Please export SPARK_HOME to the Spark path"
+  exit 1
+fi
+
 if [[ -z ${SCRIPT} ]]; then
     echo "Please select the implementation with: export SCRIPT=sparkrapids-xgboost-<implementation>.py"
     exit 1