From 354d1184076373624b6d69537694c78d578d3d99 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Mon, 30 Jun 2025 13:40:19 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 lerobot/scripts/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lerobot/scripts/train.py b/lerobot/scripts/train.py
index 9977dc931..3f109168e 100644
--- a/lerobot/scripts/train.py
+++ b/lerobot/scripts/train.py
@@ -132,7 +132,7 @@ def get_default_peft_configuration(policy_type):
 
 
 def wrap_policy_in_peft_model(cfg, policy):
-    from peft import get_peft_model, PEFT_TYPE_TO_CONFIG_MAPPING, PeftType
+    from peft import PEFT_TYPE_TO_CONFIG_MAPPING, PeftType, get_peft_model
 
     # Disable all gradients because we'll only train the parameters selected by the PEFT method.
     # Layers that should receive gradients anyway need to be listed in `modules_to_save`.