From aa805b51c412d75e9a128f450278b832b5f4d397 Mon Sep 17 00:00:00 2001
From: galipremsagar <sagarprem75@gmail.com>
Date: Wed, 27 Mar 2024 11:45:53 +0000
Subject: [PATCH] Enable all tests for arm builds

---
 ci/test_wheel_cudf.sh | 24 +++++++++---------------
 1 file changed, 9 insertions(+), 15 deletions(-)

diff --git a/ci/test_wheel_cudf.sh b/ci/test_wheel_cudf.sh
index 83f0b976128..a3211495afb 100755
--- a/ci/test_wheel_cudf.sh
+++ b/ci/test_wheel_cudf.sh
@@ -13,18 +13,12 @@ RESULTS_DIR=${RAPIDS_TESTS_DIR:-"$(mktemp -d)"}
 RAPIDS_TESTS_DIR=${RAPIDS_TESTS_DIR:-"${RESULTS_DIR}/test-results"}/
 mkdir -p "${RAPIDS_TESTS_DIR}"
 
-# Run smoke tests for aarch64 pull requests
-if [[ "$(arch)" == "aarch64" && ${RAPIDS_BUILD_TYPE} == "pull-request" ]]; then
-    rapids-logger "Run smoke tests for cudf"
-    python ./ci/wheel_smoke_test_cudf.py
-else
-    rapids-logger "pytest cudf"
-    pushd python/cudf/cudf/tests
-    python -m pytest \
-      --cache-clear \
-      --junitxml="${RAPIDS_TESTS_DIR}/junit-cudf.xml" \
-      --numprocesses=8 \
-      --dist=worksteal \
-      .
-    popd
-fi
+rapids-logger "pytest cudf"
+pushd python/cudf/cudf/tests
+python -m pytest \
+  --cache-clear \
+  --junitxml="${RAPIDS_TESTS_DIR}/junit-cudf.xml" \
+  --numprocesses=8 \
+  --dist=worksteal \
+  .
+popd