nixpkgs mirror (for testing) github.com/NixOS/nixpkgs
nix
fork

Configure Feed

Select the types of activity you want to include in your feed.

python3Packages.lerobot: init at 0.3.3

+147
+145
pkgs/development/python-modules/lerobot/default.nix
··· 1 + { 2 + lib, 3 + buildPythonPackage, 4 + fetchFromGitHub, 5 + 6 + # build-system 7 + cmake, 8 + setuptools, 9 + 10 + # dependencies 11 + av, 12 + datasets, 13 + deepdiff, 14 + diffusers, 15 + draccus, 16 + einops, 17 + flask, 18 + gymnasium, 19 + huggingface-hub, 20 + imageio, 21 + jsonlines, 22 + opencv-python-headless, 23 + packaging, 24 + pynput, 25 + pyserial, 26 + rerun-sdk, 27 + termcolor, 28 + torch, 29 + torchvision, 30 + wandb, 31 + 32 + # tests 33 + pytestCheckHook, 34 + writableTmpDirAsHomeHook, 35 + }: 36 + 37 + buildPythonPackage rec { 38 + pname = "lerobot"; 39 + version = "0.3.3"; 40 + pyproject = true; 41 + 42 + src = fetchFromGitHub { 43 + owner = "huggingface"; 44 + repo = "lerobot"; 45 + tag = "v${version}"; 46 + hash = "sha256-JaspLUuSupmk6QD+mUVLuCSczTNjy0w+nsLBcy7tXnE="; 47 + }; 48 + 49 + # ValueError: mutable default <class 'lerobot.configs.types.PolicyFeature'> for field value is not allowed: use default_factory 50 + postPatch = '' 51 + substituteInPlace tests/processor/test_pipeline.py \ 52 + --replace-fail \ 53 + "from dataclasses import dataclass" \ 54 + "from dataclasses import dataclass, field" \ 55 + --replace-fail \ 56 + "value: PolicyFeature = PolicyFeature(type=FeatureType.STATE, shape=(1,))" \ 57 + "value: PolicyFeature = field(default_factory=lambda: PolicyFeature(type=FeatureType.STATE, shape=(1,)))" 58 + ''; 59 + 60 + build-system = [ 61 + cmake 62 + setuptools 63 + ]; 64 + dontUseCmakeConfigure = true; 65 + 66 + pythonRelaxDeps = [ 67 + "datasets" 68 + "draccus" 69 + "gymnasium" 70 + "rerun-sdk" 71 + "torch" 72 + "torchvision" 73 + ]; 74 + 75 + dependencies = [ 76 + av 77 + datasets 78 + deepdiff 79 + diffusers 80 + draccus 81 + einops 82 + flask 83 + gymnasium 84 + huggingface-hub 85 + imageio 86 + jsonlines 87 + opencv-python-headless 88 + packaging 89 + pynput 90 + pyserial 91 + rerun-sdk 92 + termcolor 93 + torch 94 + torchvision 95 + wandb 96 + ]; 97 + 98 + pythonImportsCheck = [ "lerobot" ]; 99 + 100 + nativeCheckInputs = [ 101 + pytestCheckHook 102 + writableTmpDirAsHomeHook 103 + ]; 104 + 105 + disabledTests = [ 106 + # RuntimeError: OpenCVCamera(/build/source/tests/artifacts/cameras/image_480x270.png) read failed 107 + "test_async_read" 108 + "test_read" 109 + "test_rotation" 110 + 111 + # Require internet access 112 + "test_act_backbone_lr" 113 + "test_backward_compatibility" 114 + "test_dataset_initialization" 115 + "test_factory" 116 + "test_policy_defaults" 117 + "test_save_and_load_pretrained" 118 + 119 + # TypeError: stack(): argument 'tensors' (position 1) must be tuple of Tensors, not Column 120 + "test_check_timestamps_sync_slightly_off" 121 + "test_check_timestamps_sync_synced" 122 + "test_check_timestamps_sync_unsynced" 123 + "test_check_timestamps_sync_unsynced_no_exception" 124 + "test_compute_sampler_weights_drop_n_last_frames" 125 + "test_compute_sampler_weights_nontrivial_ratio" 126 + "test_compute_sampler_weights_nontrivial_ratio_and_drop_last_n" 127 + "test_compute_sampler_weights_trivial" 128 + "test_record_and_replay" 129 + "test_record_and_resume" 130 + "test_same_attributes_defined" 131 + 132 + # AssertionError between two dicts. One has an extra `'_is_initial': False` entry. 133 + "test_cosine_decay_with_warmup_scheduler" 134 + "test_diffuser_scheduler" 135 + "test_vqbet_scheduler" 136 + ]; 137 + 138 + meta = { 139 + description = "Making AI for Robotics more accessible with end-to-end learning"; 140 + homepage = "https://github.com/huggingface/lerobot"; 141 + changelog = "https://github.com/huggingface/lerobot/releases/tag/v${version}"; 142 + license = lib.licenses.asl20; 143 + maintainers = with lib.maintainers; [ GaetanLepage ]; 144 + }; 145 + }
+2
pkgs/top-level/python-packages.nix
··· 8135 8135 8136 8136 lektricowifi = callPackage ../development/python-modules/lektricowifi { }; 8137 8137 8138 + lerobot = callPackage ../development/python-modules/lerobot { }; 8139 + 8138 8140 letpot = callPackage ../development/python-modules/letpot { }; 8139 8141 8140 8142 leveldb = callPackage ../development/python-modules/leveldb { };