From e91e48b79c81953f2a8b7f5387c7391cbc9c5b79 Mon Sep 17 00:00:00 2001 From: "jade.choghari@huggingface.co" <“chogharijade@gmail.com”> Date: Tue, 18 Nov 2025 14:25:48 +0100 Subject: [PATCH] fix style --- tests/processor/test_libero_processor.py | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/tests/processor/test_libero_processor.py b/tests/processor/test_libero_processor.py index fa28e850d..14f7d9651 100644 --- a/tests/processor/test_libero_processor.py +++ b/tests/processor/test_libero_processor.py @@ -15,17 +15,19 @@ # limitations under the License. import numpy as np -from lerobot.envs.utils import preprocess_observation -from lerobot.processor.pipeline import PolicyProcessorPipeline -from lerobot.processor.observation_processor import LiberoProcessorStep import torch + +from lerobot.envs.utils import preprocess_observation +from lerobot.processor.observation_processor import LiberoProcessorStep +from lerobot.processor.pipeline import PolicyProcessorPipeline + seed = 42 np.random.seed(seed) B = 5 obs1 = { "pixels": { - "image": (np.random.rand(B, 256, 256, 3) * 255).astype(np.uint8), + "image": (np.random.rand(B, 256, 256, 3) * 255).astype(np.uint8), "image2": (np.random.rand(B, 256, 256, 3) * 255).astype(np.uint8), }, "robot_state": { @@ -42,8 +44,8 @@ obs1 = { "joints": { "pos": np.random.randn(B, 7), "vel": np.random.randn(B, 7), - } - } + }, + }, } observation = preprocess_observation(obs1)