diff --git a/pkgs/development/python-modules/pytorch-lightning/default.nix b/pkgs/development/python-modules/pytorch-lightning/default.nix index d3c9a965515..4e32226fe79 100644 --- a/pkgs/development/python-modules/pytorch-lightning/default.nix +++ b/pkgs/development/python-modules/pytorch-lightning/default.nix @@ -3,10 +3,13 @@ , fetchFromGitHub , isPy27 , future +, fsspec +, packaging , pytestCheckHook , pytorch , pyyaml , tensorboard +, torchmetrics , tqdm }: buildPythonPackage rec { @@ -22,11 +25,19 @@ buildPythonPackage rec { sha256 = "sha256-GP6/VZuRv8dS5wKQW7RbtOSa2vV9Af2Jp+ioEW3bIgc="; }; + postPatch = '' + sed -i '/setuptools==/d' requirements.txt + sed -i 's/pyDeprecate==/pyDeprecate>=/' requirements.txt + ''; + propagatedBuildInputs = [ + packaging future + fsspec pytorch pyyaml tensorboard + torchmetrics tqdm ]; diff --git a/pkgs/development/python-modules/torchmetrics/default.nix b/pkgs/development/python-modules/torchmetrics/default.nix index fc2f6cf7535..ddccb4d7865 100644 --- a/pkgs/development/python-modules/torchmetrics/default.nix +++ b/pkgs/development/python-modules/torchmetrics/default.nix @@ -10,6 +10,7 @@ , pytorch , pytestCheckHook , torchmetrics +, pytorch-lightning }: let @@ -37,6 +38,7 @@ buildPythonPackage { ]; checkInputs = [ + pytorch-lightning scikit-learn scikitimage cloudpickle