diff --git a/lerobot/record.py b/lerobot/record.py index 01f835718..506372eea 100644 --- a/lerobot/record.py +++ b/lerobot/record.py @@ -155,7 +155,7 @@ class RecordConfig: self.policy = PreTrainedConfig.from_pretrained(policy_path, cli_overrides=cli_overrides) self.policy.pretrained_path = policy_path - if (Path(policy_path) / 'adapter_config.json').exists(): + if (Path(policy_path) / "adapter_config.json").exists(): self.policy.use_peft = True if self.teleop is None and self.policy is None: diff --git a/lerobot/scripts/train.py b/lerobot/scripts/train.py index 8f172d0ea..9977dc931 100644 --- a/lerobot/scripts/train.py +++ b/lerobot/scripts/train.py @@ -128,7 +128,7 @@ def get_default_peft_configuration(policy_type): ], } - return {'modules_to_save': None} + return {"modules_to_save": None} def wrap_policy_in_peft_model(cfg, policy): @@ -149,7 +149,7 @@ def wrap_policy_in_peft_model(cfg, policy): if peft_config_cli[key] is not None: peft_config_policy[key] = peft_config_cli[key] - if 'target_modules' not in peft_config_policy: + if "target_modules" not in peft_config_policy: raise ValueError( f"There is no default `target_modules` value for policy {cfg.policy.type}. Please pass it manually." )