From af1760f1757aa1436716422264acaeac11e7e320 Mon Sep 17 00:00:00 2001
From: Steven Palma <imstevenpmwork@ieee.org>
Date: Wed, 24 Sep 2025 16:46:38 +0200
Subject: [PATCH] chore(utils): move benchmark and buffer to their respective
 modules (#2028)

---
 {src/lerobot/utils => benchmarks/video}/benchmark.py | 0
 benchmarks/video/run_video_benchmark.py              | 2 +-
 src/lerobot/{utils => rl}/buffer.py                  | 0
 src/lerobot/rl/learner.py                            | 2 +-
 tests/utils/test_replay_buffer.py                    | 2 +-
 5 files changed, 3 insertions(+), 3 deletions(-)
 rename {src/lerobot/utils => benchmarks/video}/benchmark.py (100%)
 rename src/lerobot/{utils => rl}/buffer.py (100%)

diff --git a/src/lerobot/utils/benchmark.py b/benchmarks/video/benchmark.py
similarity index 100%
rename from src/lerobot/utils/benchmark.py
rename to benchmarks/video/benchmark.py
diff --git a/benchmarks/video/run_video_benchmark.py b/benchmarks/video/run_video_benchmark.py
index 5472551f..f041a906 100644
--- a/benchmarks/video/run_video_benchmark.py
+++ b/benchmarks/video/run_video_benchmark.py
@@ -35,12 +35,12 @@ import torch
 from skimage.metrics import mean_squared_error, peak_signal_noise_ratio, structural_similarity
 from tqdm import tqdm
 
+from benchmarks.video.benchmark import TimeBenchmark
 from lerobot.datasets.lerobot_dataset import LeRobotDataset
 from lerobot.datasets.video_utils import (
     decode_video_frames_torchvision,
     encode_video_frames,
 )
-from lerobot.utils.benchmark import TimeBenchmark
 
 BASE_ENCODING = OrderedDict(
     [
diff --git a/src/lerobot/utils/buffer.py b/src/lerobot/rl/buffer.py
similarity index 100%
rename from src/lerobot/utils/buffer.py
rename to src/lerobot/rl/buffer.py
diff --git a/src/lerobot/rl/learner.py b/src/lerobot/rl/learner.py
index 8d683128..6fd9fb86 100644
--- a/src/lerobot/rl/learner.py
+++ b/src/lerobot/rl/learner.py
@@ -66,6 +66,7 @@ from lerobot.datasets.factory import make_dataset
 from lerobot.datasets.lerobot_dataset import LeRobotDataset
 from lerobot.policies.factory import make_policy
 from lerobot.policies.sac.modeling_sac import SACPolicy
+from lerobot.rl.buffer import ReplayBuffer, concatenate_batch_transitions
 from lerobot.robots import so100_follower  # noqa: F401
 from lerobot.teleoperators import gamepad, so101_leader  # noqa: F401
 from lerobot.teleoperators.utils import TeleopEvents
@@ -76,7 +77,6 @@ from lerobot.transport.utils import (
     bytes_to_transitions,
     state_to_bytes,
 )
-from lerobot.utils.buffer import ReplayBuffer, concatenate_batch_transitions
 from lerobot.utils.constants import (
     CHECKPOINTS_DIR,
     LAST_CHECKPOINT_LINK,
diff --git a/tests/utils/test_replay_buffer.py b/tests/utils/test_replay_buffer.py
index 8781c5c0..b5254f39 100644
--- a/tests/utils/test_replay_buffer.py
+++ b/tests/utils/test_replay_buffer.py
@@ -21,7 +21,7 @@ import pytest
 import torch
 
 from lerobot.datasets.lerobot_dataset import LeRobotDataset
-from lerobot.utils.buffer import BatchTransition, ReplayBuffer, random_crop_vectorized
+from lerobot.rl.buffer import BatchTransition, ReplayBuffer, random_crop_vectorized
 from tests.fixtures.constants import DUMMY_REPO_ID