From 329bdb57026e11ee9c51621ac73daaaae2aa2a0c Mon Sep 17 00:00:00 2001
From: Peter Andreas Entschev <peter@entschev.com>
Date: Wed, 26 Jun 2024 14:41:30 -0700
Subject: [PATCH] Fix `ci/test_python.sh` arguments

---
 ci/test_python.sh | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/ci/test_python.sh b/ci/test_python.sh
index 5bb24d11..78330a40 100755
--- a/ci/test_python.sh
+++ b/ci/test_python.sh
@@ -136,7 +136,7 @@ DASK_DATAFRAME__QUERY_PLANNING=True \
 python dask_cuda/benchmarks/local_cudf_shuffle.py \
   --rmm-pool-size 2GiB \
   --rmm-maximum-pool-size 4GiB \
-  --rmm-async \
+  --enable-rmm-async \
   --partition-size="1 KiB" \
   -d 0 \
   --runs 1 \
@@ -146,7 +146,7 @@ DASK_DATAFRAME__QUERY_PLANNING=True \
 python dask_cuda/benchmarks/local_cudf_shuffle.py \
   --rmm-pool-size 2GiB \
   --rmm-maximum-pool-size 4GiB \
-  --rmm-managed \
+  --enable-rmm-managed \
   --partition-size="1 KiB" \
   -d 0 \
   --runs 1 \