[console_scripts]
trainnr = trainnr.cli:main

[trainnr.engines]
mjx-warp = trainnr.physics.mjx_backend
mujoco = trainnr.physics.mujoco_backend

[trainnr.gpu_providers]
runpod = trainnr.cloud.runpod

[trainnr.identification_methods]
drivetrain-ratio = trainnr.robot.methods:DrivetrainRatio

[trainnr.model_sources]
usd = trainnr.robot.usd_import

[trainnr.robot_adapters]
lerobot = trainnr.robots.adapters.lerobot
mcap = trainnr.robots.adapters.mcap
rosbag2 = trainnr.robots.adapters.rosbag2
wire = trainnr.robots.adapters.wire

[trainnr.tasks]
aloha2 = trainnr.tasks.aloha2
so101 = trainnr.tasks.so101
