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

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

2
.gitignore vendored
View File

@@ -173,4 +173,4 @@ dmypy.json
cython_debug/
# for mac
._*
._*

View File

@@ -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

View File

@@ -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(