mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-29 20:49:42 +00:00
refactor(unitree_g1): minimize diff vs main (drop onnx guards, e-stop)
Revert pyproject.toml and import_utils.py to main: the onnxruntime-gpu detection fallback and the _onnxruntime_available/_onnx_available flags are unnecessary. Controllers now import onnx/onnxruntime unconditionally (matching main) which also works with onnxruntime-gpu since the import name is stable. Drop the require_package() calls we had added. Remove the stdin e-stop listener from serve_onboard_controller and the now-unused os/sys imports; Ctrl-C still triggers graceful shutdown.
This commit is contained in:
+1
-5
@@ -374,11 +374,7 @@ torch = [{ index = "pytorch-cu128", marker = "sys_platform == 'linux'" }]
|
||||
torchvision = [{ index = "pytorch-cu128", marker = "sys_platform == 'linux'" }]
|
||||
|
||||
[tool.setuptools.package-data]
|
||||
lerobot = [
|
||||
"envs/*.json",
|
||||
"annotations/steerable_pipeline/prompts/*.txt",
|
||||
"teleoperators/pico_headset/assets/*.npz",
|
||||
]
|
||||
lerobot = ["envs/*.json", "annotations/steerable_pipeline/prompts/*.txt"]
|
||||
|
||||
[tool.setuptools.packages.find]
|
||||
where = ["src"]
|
||||
|
||||
Reference in New Issue
Block a user