hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse Source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot]
2025-05-30 11:14:46 +00:00
parent a3a73f6822
commit ac0f2e9dd9
3 changed files with 7 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
examples/train_smolvla.sh
View File

@@ -103,4 +103,3 @@ python lerobot/scripts/train.py \
--output_dir=$TRAIN_DIR \ --output_dir=$TRAIN_DIR \
--batch_size=$BATCH_SIZE \ --batch_size=$BATCH_SIZE \
--steps=$OFFLINE_STEPS --steps=$OFFLINE_STEPS

2
lerobot/common/robot_devices/control_utils.py
View File

@@ -256,7 +256,7 @@ def control_loop(
else: else:
observation = robot.capture_observation() observation = robot.capture_observation()
action = None action = None
observation['task'] = [single_task] observation["task"] = [single_task]
observation["robot_type"] = [policy.robot_type] if hasattr(policy, "robot_type") else [""] observation["robot_type"] = [policy.robot_type] if hasattr(policy, "robot_type") else [""]
if policy is not None: if policy is not None:
pred_action = predict_action( pred_action = predict_action(