diff --git a/pkgs/development/python-modules/lightning/default.nix b/pkgs/development/python-modules/lightning/default.nix new file mode 100644 index 000000000000..e3a096c9163b --- /dev/null +++ b/pkgs/development/python-modules/lightning/default.nix @@ -0,0 +1,43 @@ +{ + 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" + ]; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 7f725c55c9c1..57d880a4b71b 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -7869,6 +7869,8 @@ self: super: with self; { lightify = callPackage ../development/python-modules/lightify { }; + lightning = callPackage ../development/python-modules/lightning { }; + lightning-utilities = callPackage ../development/python-modules/lightning-utilities { }; lightparam = callPackage ../development/python-modules/lightparam { };