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"
|
|
];
|
|
}
|