diff --git a/pkgs/development/python-modules/lerobot/default.nix b/pkgs/development/python-modules/lerobot/default.nix index 675833d6ec06..0cfd75ad5c6b 100644 --- a/pkgs/development/python-modules/lerobot/default.nix +++ b/pkgs/development/python-modules/lerobot/default.nix @@ -60,6 +60,7 @@ buildPythonPackage (finalAttrs: { "datasets" "draccus" "gymnasium" + "huggingface-hub" "rerun-sdk" "torch" "torchvision" @@ -91,9 +92,7 @@ buildPythonPackage (finalAttrs: { torchvision wandb ] - ++ imageio.optional-dependencies.ffmpeg - ++ huggingface-hub.optional-dependencies.hf_transfer - ++ huggingface-hub.optional-dependencies.cli; + ++ imageio.optional-dependencies.ffmpeg; pythonImportsCheck = [ "lerobot" ];