From 88ccbb6b1f426bedebade4fd00bd5c95d9cd2d75 Mon Sep 17 00:00:00 2001
From: "Rishi C." <77904151+rishic3@users.noreply.github.com>
Date: Tue, 19 Nov 2024 10:27:05 -0800
Subject: [PATCH] Update run-sparkrapids-xgboost.sh

---
 .../optuna-examples/standalone/run-sparkrapids-xgboost.sh    | 5 +++++
 1 file changed, 5 insertions(+)

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-<implementation>.py"
     exit 1