summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorJiajie Chen <c@jia.je>2022-11-01 20:59:35 +0800
committerMatthias Beyer <mail@beyermatthias.de>2023-01-09 15:20:53 +0100
commit522273064f80d0142ba0bedd6b124c1b14e5f736 (patch)
tree302ab1fa77988b963081bc06e8e06ff493ce8bfa
parent5ab9d06c948d660bb8adb541a8e2588ac2f8a1ea (diff)
python3Packages.pytorch-lightning: 1.6.5 -> 1.8.0
-rw-r--r--pkgs/development/python-modules/pytorch-lightning/default.nix31
1 files changed, 10 insertions, 21 deletions
diff --git a/pkgs/development/python-modules/pytorch-lightning/default.nix b/pkgs/development/python-modules/pytorch-lightning/default.nix
index a2d8da399d28..0be1967a4520 100644
--- a/pkgs/development/python-modules/pytorch-lightning/default.nix
+++ b/pkgs/development/python-modules/pytorch-lightning/default.nix
@@ -1,39 +1,28 @@
{ lib
, buildPythonPackage
-, fetchFromGitHub
+, fetchPypi
, isPy27
-, future
-, fsspec
-, packaging
, pytestCheckHook
-, torch
-, pyyaml
-, tensorboard
+, lightning-lite
+, lightning-utilities
, torchmetrics
-, tqdm }:
+}:
buildPythonPackage rec {
pname = "pytorch-lightning";
- version = "1.6.5";
+ version = "1.8.0";
disabled = isPy27;
- src = fetchFromGitHub {
- owner = "PyTorchLightning";
- repo = pname;
- rev = "refs/tags/${version}";
- hash = "sha256-CgD5g5nhz2DI4gOQyPl8/Cq6wWHzL0ALgOB5SgUOgaI=";
+ src = fetchPypi {
+ inherit pname version;
+ hash = "sha256-3v+bx5eOzryPReiBre9l3I2fRVTojDsGT4BYfzKrFY0=";
};
propagatedBuildInputs = [
- packaging
- future
- fsspec
- torch
- pyyaml
- tensorboard
+ lightning-lite
+ lightning-utilities
torchmetrics
- tqdm
];
checkInputs = [ pytestCheckHook ];