From 623412406eb4a5cf1358568e02153eb2edc4b6b2 Mon Sep 17 00:00:00 2001
From: David Brazda <davidbrazda61@gmail.com>
Date: Wed, 27 Nov 2024 10:15:22 +0100
Subject: [PATCH] fix

---
 setup.py         | 2 +-
 ttools/models.py | 4 +++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index 298c033..bb8fbbe 100644
--- a/setup.py
+++ b/setup.py
@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='ttools',
-    version='0.7.93',
+    version='0.7.94',
     packages=find_packages(),
     install_requires=[
         # list your dependencies here
diff --git a/ttools/models.py b/ttools/models.py
index 2e0c28e..2aab27a 100644
--- a/ttools/models.py
+++ b/ttools/models.py
@@ -902,7 +902,9 @@ class LibraryTradingModel:
                         model.set_params(num_class=self.config.n_classes)
                 else:
                     model = XGBRegressor(**params)
-                
+
+                model = set_gpu_params(model)
+
                 # Handle class imbalance for binary classification
                 if self.config.n_classes == 2:
                     n_0 = sum(y_fold_train == 0)