From b859e89936523f732b4f797183a06af553c51b5d Mon Sep 17 00:00:00 2001
From: Simon Alibert <alibert.sim@gmail.com>
Date: Sun, 3 Mar 2024 13:05:21 +0100
Subject: [PATCH] Fix for PR #5

---
 lerobot/scripts/train.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/lerobot/scripts/train.py b/lerobot/scripts/train.py
index c2eef313..3e0b2a9a 100644
--- a/lerobot/scripts/train.py
+++ b/lerobot/scripts/train.py
@@ -1,5 +1,4 @@
 import logging
-import warnings
 
 import hydra
 import numpy as np
@@ -119,7 +118,7 @@ def train(cfg: dict, out_dir=None, job_name=None):
     if cfg.device == "cuda":
         assert torch.cuda.is_available()
     else:
-        warnings.warn("Using CPU, this will be slow.", UserWarning, stacklevel=1)
+        logging.warning("Using CPU, this will be slow.")
 
     torch.backends.cudnn.benchmark = True
     torch.backends.cuda.matmul.allow_tf32 = True