Move so100_dataset into lerobot package, remove PYTHONPATH hack

lerobot/src/lerobot/datasets/factory.py

@@ -85,7 +85,7 @@ def make_dataset(cfg: TrainPipelineConfig) -> LeRobotDataset | MultiLeRobotDatas
     # Support SO100Dataset via repo_id starting with "so100:"
     # Format: "so100:/path/to/data_root:/path/to/index.json:/path/to/stats.json"
     if isinstance(cfg.dataset.repo_id, str) and cfg.dataset.repo_id.startswith("so100:"):
-        from so100_dataset import SO100Dataset
+        from lerobot.datasets.so100_dataset import SO100Dataset
 
         parts = cfg.dataset.repo_id.split(":")
         data_root = parts[1]

train_cloud.sh

@@ -29,7 +29,6 @@ if [ "$NUM_GPUS" -gt 1 ]; then
     ACCEL_FLAGS="--multi_gpu --num_processes $NUM_GPUS"
 fi
 
-PYTHONPATH=/workspace/pi05-so100-diverse:$PYTHONPATH \
 accelerate launch $ACCEL_FLAGS \
 -m lerobot.scripts.lerobot_train \
 --dataset.repo_id="so100:$DATASET_DIR:/workspace/pi05-so100-diverse/filtered_index.json:/workspace/pi05-so100-diverse/norm_stats.json" \