When importing with "import lerobot.common.datasets.lerobot_dataset as lerobot_dataset", the process crashes.

[JYounngS]

I want to fine-tune Pi0.5 on my own data. But after I ran "XLA_PYTHON_CLIENT_MEM_FRACTION=0.9 uv run scripts/train.py pi05_aloha --exp-name=my_experiment --overwrite", the process exits silently (with no error message).

Then I found that the issue occurs at the line in src/openpi/training/data_loader.py where it imports lerobot.common.datasets.lerobot_dataset as lerobot_dataset.

However, I was able to import lerobot.common.datasets.lerobot_dataset as lerobot_dataset successfully in the Python interpreter launched by uv.

This issue is a bit strange. Your support will be greatly appreciated.