From ac0f2e9dd9df4506bb73c3f7f9ef7b62637ac726 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Fri, 30 May 2025 11:14:46 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- .gitignore | 2 +- examples/train_smolvla.sh | 11 +++++------ lerobot/common/robot_devices/control_utils.py | 2 +- 3 files changed, 7 insertions(+), 8 deletions(-) diff --git a/.gitignore b/.gitignore index 7cff9f2e..7f3377e1 100644 --- a/.gitignore +++ b/.gitignore @@ -173,4 +173,4 @@ dmypy.json cython_debug/ # for mac -._* \ No newline at end of file +._* diff --git a/examples/train_smolvla.sh b/examples/train_smolvla.sh index 8c06aa45..6c4cf956 100644 --- a/examples/train_smolvla.sh +++ b/examples/train_smolvla.sh @@ -22,14 +22,14 @@ ##SBATCH -C a100 ###SBATCH -A dyf@a100 ##SBATCH -A lqm@a100 -##SBATCH --qos=qos_gpu_a100-dev -##SBATCH --qos=qos_gpu_a100-t3 +##SBATCH --qos=qos_gpu_a100-dev +##SBATCH --qos=qos_gpu_a100-t3 #SBATCH --partition=gpu_p6 #SBATCH -C h100 #SBATCH -A lqm@h100 -##SBATCH --qos=qos_gpu_h100-dev -#SBATCH --qos=qos_gpu_h100-t4 +##SBATCH --qos=qos_gpu_h100-dev +#SBATCH --qos=qos_gpu_h100-t4 cd ~/lerobot_pi @@ -45,7 +45,7 @@ export DATA_DIR=$WORK/.cache/huggingface/datasets export HF_LEROBOT_HOME=$WORK/.cache/huggingface/lerobot # export LEROBOT_HOME= -export HF_DATASETS_OFFLINE=1 +export HF_DATASETS_OFFLINE=1 export HF_HUB_OFFLINE=1 export WANDB_CACHE_DIR=/lustre/fsn1/projects/rech/dyf/ugz83ue/wandb @@ -103,4 +103,3 @@ python lerobot/scripts/train.py \ --output_dir=$TRAIN_DIR \ --batch_size=$BATCH_SIZE \ --steps=$OFFLINE_STEPS - diff --git a/lerobot/common/robot_devices/control_utils.py b/lerobot/common/robot_devices/control_utils.py index 103c4fdb..60656196 100644 --- a/lerobot/common/robot_devices/control_utils.py +++ b/lerobot/common/robot_devices/control_utils.py @@ -256,7 +256,7 @@ def control_loop( else: observation = robot.capture_observation() action = None - observation['task'] = [single_task] + observation["task"] = [single_task] observation["robot_type"] = [policy.robot_type] if hasattr(policy, "robot_type") else [""] if policy is not None: pred_action = predict_action(