diff --git a/pkgs/development/python-modules/gluonts/default.nix b/pkgs/development/python-modules/gluonts/default.nix index 8ec0ec353c85..0335550d148b 100644 --- a/pkgs/development/python-modules/gluonts/default.nix +++ b/pkgs/development/python-modules/gluonts/default.nix @@ -3,6 +3,9 @@ lib, buildPythonPackage, fetchFromGitHub, + fetchpatch, + + # build-system setuptools, # dependencies @@ -23,6 +26,7 @@ matplotlib, pyarrow, statsmodels, + writableTmpDirAsHomeHook, which, }: @@ -38,6 +42,16 @@ buildPythonPackage rec { hash = "sha256-h0+RYgGMz0gPchiKGIu0/NGcWBky5AWNTJKzoupn/iQ="; }; + patches = [ + # Fixes _pickle.UnpicklingError: Weights only load failed. + # https://github.com/awslabs/gluonts/pull/3269 + (fetchpatch { + name = "fix-torch-load_from_checkpoint"; + url = "https://github.com/awslabs/gluonts/pull/3269/commits/6420e75cfbeabcd94e2ff09dfed3b2eeb4881710.patch"; + hash = "sha256-UeLjgKra+Y3uPoTBle+YCxD0a1ahu6d5anrMHn4HH2I="; + }) + ]; + build-system = [ setuptools ]; @@ -83,12 +97,11 @@ buildPythonPackage rec { matplotlib pyarrow statsmodels + writableTmpDirAsHomeHook which ] ++ optional-dependencies.torch; - preCheck = ''export HOME=$(mktemp -d)''; - disabledTestPaths = [ # requires `cpflows`, not in Nixpkgs "test/torch/model" diff --git a/pkgs/development/python-modules/pytorch-lightning/default.nix b/pkgs/development/python-modules/pytorch-lightning/default.nix index 979f6c4b0fa8..fd46a69c14ef 100644 --- a/pkgs/development/python-modules/pytorch-lightning/default.nix +++ b/pkgs/development/python-modules/pytorch-lightning/default.nix @@ -24,14 +24,14 @@ buildPythonPackage rec { pname = "pytorch-lightning"; - version = "2.5.6"; + version = "2.6.0"; pyproject = true; src = fetchFromGitHub { owner = "Lightning-AI"; repo = "pytorch-lightning"; tag = version; - hash = "sha256-ojmE0d6Wy4UqQu4kBBE2qtQ4AYqplHOB7wJ7hEte664="; + hash = "sha256-zmFA9/tz0C06LmQ37wHeoR1kBHKdoz/D1cKWMoeWHzs="; }; preConfigure = ''