python3Packages.pytorch-lightning: 2.5.6 -> 2.6.0 (#466039)

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"

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 = ''