Fix GPU detection

ThomasMeissnerDS · Jun 26, 2024 · 7028e47 · 7028e47
1 parent b477d01
commit 7028e47
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bluecast/general_utils/general_utils.py b/bluecast/general_utils/general_utils.py
@@ -54,7 +54,7 @@ def check_gpu_support() -> Dict[str, str]:
         logging.info(f"Can use {params}.")
         return params
     except Exception:
-        params = {"tree_method": "hist", "device": "cpu"}
+        params = {"tree_method": "exact", "device": "cpu"}
         logging.info("Xgboost uses CPU.")
         logging.info(
             f"""Can use {params} for Xgboost (Will only be used when conf_training.autotune_on_device either