146 lines
3.3 KiB
Nix
146 lines
3.3 KiB
Nix
{
|
|
lib,
|
|
buildPythonPackage,
|
|
fetchFromGitHub,
|
|
|
|
# build-system
|
|
cmake,
|
|
setuptools,
|
|
|
|
# dependencies
|
|
av,
|
|
datasets,
|
|
deepdiff,
|
|
diffusers,
|
|
draccus,
|
|
einops,
|
|
flask,
|
|
gymnasium,
|
|
huggingface-hub,
|
|
imageio,
|
|
jsonlines,
|
|
opencv-python-headless,
|
|
packaging,
|
|
pynput,
|
|
pyserial,
|
|
rerun-sdk,
|
|
termcolor,
|
|
torch,
|
|
torchvision,
|
|
wandb,
|
|
|
|
# tests
|
|
pytestCheckHook,
|
|
writableTmpDirAsHomeHook,
|
|
}:
|
|
|
|
buildPythonPackage rec {
|
|
pname = "lerobot";
|
|
version = "0.3.3";
|
|
pyproject = true;
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "huggingface";
|
|
repo = "lerobot";
|
|
tag = "v${version}";
|
|
hash = "sha256-JaspLUuSupmk6QD+mUVLuCSczTNjy0w+nsLBcy7tXnE=";
|
|
};
|
|
|
|
# ValueError: mutable default <class 'lerobot.configs.types.PolicyFeature'> for field value is not allowed: use default_factory
|
|
postPatch = ''
|
|
substituteInPlace tests/processor/test_pipeline.py \
|
|
--replace-fail \
|
|
"from dataclasses import dataclass" \
|
|
"from dataclasses import dataclass, field" \
|
|
--replace-fail \
|
|
"value: PolicyFeature = PolicyFeature(type=FeatureType.STATE, shape=(1,))" \
|
|
"value: PolicyFeature = field(default_factory=lambda: PolicyFeature(type=FeatureType.STATE, shape=(1,)))"
|
|
'';
|
|
|
|
build-system = [
|
|
cmake
|
|
setuptools
|
|
];
|
|
dontUseCmakeConfigure = true;
|
|
|
|
pythonRelaxDeps = [
|
|
"datasets"
|
|
"draccus"
|
|
"gymnasium"
|
|
"rerun-sdk"
|
|
"torch"
|
|
"torchvision"
|
|
];
|
|
|
|
dependencies = [
|
|
av
|
|
datasets
|
|
deepdiff
|
|
diffusers
|
|
draccus
|
|
einops
|
|
flask
|
|
gymnasium
|
|
huggingface-hub
|
|
imageio
|
|
jsonlines
|
|
opencv-python-headless
|
|
packaging
|
|
pynput
|
|
pyserial
|
|
rerun-sdk
|
|
termcolor
|
|
torch
|
|
torchvision
|
|
wandb
|
|
];
|
|
|
|
pythonImportsCheck = [ "lerobot" ];
|
|
|
|
nativeCheckInputs = [
|
|
pytestCheckHook
|
|
writableTmpDirAsHomeHook
|
|
];
|
|
|
|
disabledTests = [
|
|
# RuntimeError: OpenCVCamera(/build/source/tests/artifacts/cameras/image_480x270.png) read failed
|
|
"test_async_read"
|
|
"test_read"
|
|
"test_rotation"
|
|
|
|
# Require internet access
|
|
"test_act_backbone_lr"
|
|
"test_backward_compatibility"
|
|
"test_dataset_initialization"
|
|
"test_factory"
|
|
"test_policy_defaults"
|
|
"test_save_and_load_pretrained"
|
|
|
|
# TypeError: stack(): argument 'tensors' (position 1) must be tuple of Tensors, not Column
|
|
"test_check_timestamps_sync_slightly_off"
|
|
"test_check_timestamps_sync_synced"
|
|
"test_check_timestamps_sync_unsynced"
|
|
"test_check_timestamps_sync_unsynced_no_exception"
|
|
"test_compute_sampler_weights_drop_n_last_frames"
|
|
"test_compute_sampler_weights_nontrivial_ratio"
|
|
"test_compute_sampler_weights_nontrivial_ratio_and_drop_last_n"
|
|
"test_compute_sampler_weights_trivial"
|
|
"test_record_and_replay"
|
|
"test_record_and_resume"
|
|
"test_same_attributes_defined"
|
|
|
|
# AssertionError between two dicts. One has an extra `'_is_initial': False` entry.
|
|
"test_cosine_decay_with_warmup_scheduler"
|
|
"test_diffuser_scheduler"
|
|
"test_vqbet_scheduler"
|
|
];
|
|
|
|
meta = {
|
|
description = "Making AI for Robotics more accessible with end-to-end learning";
|
|
homepage = "https://github.com/huggingface/lerobot";
|
|
changelog = "https://github.com/huggingface/lerobot/releases/tag/v${version}";
|
|
license = lib.licenses.asl20;
|
|
maintainers = with lib.maintainers; [ GaetanLepage ];
|
|
};
|
|
}
|