646b892680
Some checks failed
Periodic Merges (6h) / master → staging-nixos (push) Failing after 12m50s
Periodic Merges (6h) / master → staging-next (push) Failing after 12m54s
Periodic Merges (24h) / merge-base(master,staging) → haskell-updates (push) Failing after 11m54s
Periodic Merges (6h) / staging-next → staging (push) Failing after 12m13s
Periodic Merges (24h) / staging-next-25.05 → staging-25.05 (push) Failing after 13m24s
Periodic Merges (24h) / release-25.05 → staging-next-25.05 (push) Failing after 14m28s
44 lines
663 B
Nix
44 lines
663 B
Nix
{
|
|
lib,
|
|
buildPythonPackage,
|
|
fetchFromGitHub,
|
|
|
|
# build-system
|
|
setuptools,
|
|
|
|
# dependencies
|
|
pytorch-lightning,
|
|
|
|
# tests
|
|
psutil,
|
|
pytestCheckHook,
|
|
}:
|
|
|
|
buildPythonPackage {
|
|
pname = "lightning";
|
|
pyproject = true;
|
|
|
|
inherit (pytorch-lightning)
|
|
version
|
|
src
|
|
build-system
|
|
meta
|
|
;
|
|
|
|
dependencies = pytorch-lightning.dependencies ++ [ pytorch-lightning ];
|
|
|
|
nativeCheckInputs = [
|
|
psutil
|
|
pytestCheckHook
|
|
];
|
|
|
|
# Some packages are not in NixPkgs; other tests try to build distributed
|
|
# models, which doesn't work in the sandbox.
|
|
doCheck = false;
|
|
|
|
pythonImportsCheck = [
|
|
"lightning"
|
|
"lightning.pytorch"
|
|
];
|
|
}
|