package.nix (979B)
1 { 2 lib, 3 fetchFromGitHub, 4 python3Packages, 5 }: 6 7 python3Packages.buildPythonPackage (finalAttrs: { 8 pname = "lightly"; 9 version = "1.5.26-unstable-2026-08-14"; 10 pyproject = true; 11 12 __structuredAttrs = true; 13 14 src = fetchFromGitHub { 15 owner = "lightly-ai"; 16 repo = "lightly"; 17 rev = "483f80923c6ffd47c7f0fedab98e3d46d873a27b"; 18 hash = "sha256-Epama3wvNBhmLJSfa+AxzAh0fYRb1cQWgcK43+apvUQ="; 19 }; 20 21 build-system = with python3Packages; [ setuptools-scm ]; 22 23 dependencies = with python3Packages; [ 24 hydra-core 25 numpy 26 tqdm 27 torch 28 torchvision 29 pytorch-lightning 30 ]; 31 32 nativeCheckInputs = with python3Packages; [ 33 pytestCheckHook 34 pytest-mock 35 scikit-learn 36 ]; 37 38 disabledTests = [ 39 "test_loss_dcl" 40 ]; 41 42 meta = { 43 description = "Deep learning package for self-supervised learning"; 44 homepage = "https://github.com/lightly-ai/lightly"; 45 license = lib.licenses.mit; 46 maintainers = [ lib.maintainers.sikmir ]; 47 }; 48 })