From 0445ee50e72f51280848b82ebd6ea2c5da0d7f56 Mon Sep 17 00:00:00 2001
From: Vincent Emonet <vincent.emonet@gmail.com>
Date: Fri, 26 Jan 2024 14:26:29 +0100
Subject: [PATCH] fix

---
 src/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/train.py b/src/train.py
index 95d80ea..bef8275 100644
--- a/src/train.py
+++ b/src/train.py
@@ -391,7 +391,7 @@ def train_gpu(
         y_train, y_test = y[train_index], y[test_index]
 
         # # Send data to GPU for xgboost
-        # send_time = time.time()
+        send_time = time.time()
         # dtrain = xgb.DMatrix(X_train, label=y_train)
         # dtest = xgb.DMatrix(X_test, label=y_test)
         # print(f"Sending data to GPU took {time.time() - send_time}s")