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 index ec268e53..edd9bde6 100755 --- 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-.py" exit 1