Make memleak pytests run on nightly jobs

rapidsai · Jan 28, 2025 · 6ef8118 · 6ef8118
1 parent 6985c51
commit 6ef8118
Showing 1 changed file with 13 additions and 9 deletions.
diff --git a/ci/test_python_singlegpu.sh b/ci/test_python_singlegpu.sh
@@ -31,16 +31,20 @@ rapids-logger "pytest cuml single GPU"
   --cov-report=xml:"${RAPIDS_COVERAGE_DIR}/cuml-accel-coverage.xml" \
   --cov-report=term
 
-rapids-logger "memory leak pytests"
 
-./ci/run_cuml_singlegpu_memleak_pytests.sh \
-  --numprocesses=1 \
-  --junitxml="${RAPIDS_TESTS_DIR}/junit-cuml-memleak.xml" \
-  --cov-config=../../.coveragerc \
-  --cov=cuml \
-  --cov-report=xml:"${RAPIDS_COVERAGE_DIR}/cuml-memleak-coverage.xml" \
-  --cov-report=term \
-  -m "memleak"
+if [ "${RAPIDS_BUILD_TYPE}" == "nightly" ]; then
+  rapids-logger "memory leak pytests"
+
+  ./ci/run_cuml_singlegpu_memleak_pytests.sh \
+    --numprocesses=1 \
+    --junitxml="${RAPIDS_TESTS_DIR}/junit-cuml-memleak.xml" \
+    --cov-config=../../.coveragerc \
+    --cov=cuml \
+    --cov-report=xml:"${RAPIDS_COVERAGE_DIR}/cuml-memleak-coverage.xml" \
+    --cov-report=term \
+    -m "memleak"
+fi
 
 rapids-logger "Test script exiting with value: $EXITCODE"
 exit ${EXITCODE}
+