pyrox.dev
1{
2 lib,
3 buildPythonPackage,
4 fetchFromGitHub,
5
6 # build-system
7 setuptools,
8
9 # dependencies
10 pytorch-lightning,
11
12 # tests
13 psutil,
14 pytestCheckHook,
15}:
16
17buildPythonPackage {
18 pname = "lightning";
19 pyproject = true;
20
21 inherit (pytorch-lightning)
22 version
23 src
24 build-system
25 meta
26 ;
27
28 dependencies = pytorch-lightning.dependencies ++ [ pytorch-lightning ];
29
30 nativeCheckInputs = [
31 psutil
32 pytestCheckHook
33 ];
34
35 # Some packages are not in NixPkgs; other tests try to build distributed
36 # models, which doesn't work in the sandbox.
37 doCheck = false;
38
39 pythonImportsCheck = [
40 "lightning"
41 "lightning.pytorch"
42 ];
43}