pyrox.dev
lol
python313Packages.lightning: init at 2.5.1
+43
pkgs/development/python-modules/lightning/default.nix
+43
pkgs/development/python-modules/lightning/default.nix
···
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
+
17
+
buildPythonPackage {
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
+
}
+2
pkgs/top-level/python-packages.nix
+2
pkgs/top-level/python-packages.nix
···
7869
7869
7870
7870
lightify = callPackage ../development/python-modules/lightify { };
7871
7871
7872
+
lightning = callPackage ../development/python-modules/lightning { };
7873
+
7872
7874
lightning-utilities = callPackage ../development/python-modules/lightning-utilities { };
7873
7875
7874
7876
lightparam = callPackage ../development/python-modules/lightparam { };