pkgs/development/python-modules/lightning/default.nix


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38

{
  buildPythonPackage,

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