diff options
| author | Jiajie Chen <c@jia.je> | 2022-11-01 20:58:45 +0800 |
|---|---|---|
| committer | Matthias Beyer <mail@beyermatthias.de> | 2023-01-09 15:20:53 +0100 |
| commit | 5ab9d06c948d660bb8adb541a8e2588ac2f8a1ea (patch) | |
| tree | 4051c32f523e32b73fc657a4d4040dd10bb9e3b0 | |
| parent | f012b316faba914b3139c090bd3ee4f7e3cc1ed8 (diff) | |
python3Packages.lightning-lite: init at 1.8.0
| -rw-r--r-- | pkgs/development/python-modules/lightning-lite/default.nix | 36 | ||||
| -rw-r--r-- | pkgs/top-level/python-packages.nix | 2 |
2 files changed, 38 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/lightning-lite/default.nix b/pkgs/development/python-modules/lightning-lite/default.nix new file mode 100644 index 000000000000..48c9ff438313 --- /dev/null +++ b/pkgs/development/python-modules/lightning-lite/default.nix @@ -0,0 +1,36 @@ +{ lib +, fetchPypi +, buildPythonPackage +, packaging +, lightning-utilities +, numpy +, fsspec +, torch +}: + +buildPythonPackage rec { + pname = "lightning-lite"; + version = "1.8.0"; + + src = fetchPypi { + inherit pname version; + hash = "sha256-pxlAQJ09GluyD2NxbIanRRV84wEA8cFmAN/jPZtleVU="; + }; + + propagatedBuildInputs = [ + fsspec + lightning-utilities + numpy + packaging + torch + ]; + + pythonImportsCheck = [ "lightning_lite" ]; + + meta = with lib; { + description = "Lightweight PyTorch wrapper for machine learning researchers"; + homepage = "https://pytorch-lightning.readthedocs.io"; + license = licenses.asl20; + maintainers = with maintainers; [ jiegec ]; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index f7e95066a029..6ec14628d49c 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -5359,6 +5359,8 @@ self: super: with self; { lightning = callPackage ../development/python-modules/lightning { }; + lightning-lite = callPackage ../development/python-modules/lightning-lite { }; + lightning-utilities = callPackage ../development/python-modules/lightning-utilities { }; lightparam = callPackage ../development/python-modules/lightparam { }; |