Backport PR rapidsai#6364

Author: dantegd

python/cuml/cuml/experimental/accel/__main__.py

@@ -22,10 +22,6 @@
 import runpy
 import sys
 
-from cuml.internals import logger
-
-from . import install
-
 
 @click.command()
 @click.option("-m", "module", required=False, help="Module to run")
@@ -66,10 +62,32 @@
     count=True,
     help="Increase output verbosity (can be used multiple times, e.g. -vv). Default shows warnings only.",
 )
+@click.option(
+    "--cudf-pandas",
+    is_flag=True,
+    default=False,
+    help="Turn on cudf.pandas alongside cuml.accel.",
+)
 @click.argument("args", nargs=-1)
 def main(
-    module, convert_to_sklearn, format, output, disable_uvm, verbose, args
+    module,
+    convert_to_sklearn,
+    format,
+    output,
+    disable_uvm,
+    verbose,
+    cudf_pandas,
+    args,
 ):
+    if cudf_pandas:
+        import cudf.pandas
+
+        cudf.pandas.install()
+
+    # avoid importing cuML before cuDF.pandas
+    from cuml.internals import logger
+    from . import install
+
     default_logger_level_index = list(logger.level_enum).index(
         logger.level_enum.warn
     )

python/cuml/cuml/tests/test_pickle.py

@@ -1,4 +1,4 @@
-# Copyright (c) 2019-2024, NVIDIA CORPORATION.
+# Copyright (c) 2019-2025, NVIDIA CORPORATION.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.