From 0ccc60f20b0cd6639e5603232e6e6889126c0a04 Mon Sep 17 00:00:00 2001 From: Jade Choghari Date: Fri, 21 Nov 2025 10:44:19 +0100 Subject: [PATCH] style --- src/lerobot/policies/xvla/__init__.py | 2 +- src/lerobot/policies/xvla/processor_xvla.py | 4 +++- src/lerobot/processor/normalize_processor.py | 2 +- 3 files changed, 5 insertions(+), 3 deletions(-) diff --git a/src/lerobot/policies/xvla/__init__.py b/src/lerobot/policies/xvla/__init__.py index 2ebe51bca..42ead21df 100644 --- a/src/lerobot/policies/xvla/__init__.py +++ b/src/lerobot/policies/xvla/__init__.py @@ -1,2 +1,2 @@ # add domainid -from lerobot.policies.xvla.processor_xvla import XVLAAddDomainIdProcessorStep \ No newline at end of file +from lerobot.policies.xvla.processor_xvla import XVLAAddDomainIdProcessorStep diff --git a/src/lerobot/policies/xvla/processor_xvla.py b/src/lerobot/policies/xvla/processor_xvla.py index e3b76fb56..388f04e95 100644 --- a/src/lerobot/policies/xvla/processor_xvla.py +++ b/src/lerobot/policies/xvla/processor_xvla.py @@ -467,7 +467,9 @@ def make_xvla_libero_pre_post_processors() -> tuple[ """ pre_processor_steps: list[ProcessorStep] = [] post_processor_steps: list[ProcessorStep] = [] - pre_processor_steps.extend([LiberoProcessorStep(), XVLAImageNetNormalizeProcessorStep(), XVLAAddDomainIdProcessorStep()]) + pre_processor_steps.extend( + [LiberoProcessorStep(), XVLAImageNetNormalizeProcessorStep(), XVLAAddDomainIdProcessorStep()] + ) post_processor_steps.extend([XVLARotation6DToAxisAngleProcessorStep()]) return ( PolicyProcessorPipeline[dict[str, Any], dict[str, Any]]( diff --git a/src/lerobot/processor/normalize_processor.py b/src/lerobot/processor/normalize_processor.py index cb17bd341..368c9b270 100644 --- a/src/lerobot/processor/normalize_processor.py +++ b/src/lerobot/processor/normalize_processor.py @@ -557,4 +557,4 @@ def hotswap_stats( step.stats = stats # Re-initialize tensor_stats on the correct device. step._tensor_stats = to_tensor(stats, device=step.device, dtype=step.dtype) # type: ignore[assignment] - return rp \ No newline at end of file + return rp