From 2a586d564f7dd78dfe782e3740693907fdece945 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Fri, 28 Nov 2025 19:58:41 +0000
Subject: [PATCH 1/2] python3Packages.pytorch-lightning: 2.5.6 -> 2.6.0

Diff: https://github.com/Lightning-AI/pytorch-lightning/compare/2.5.6...2.6.0

Changelog: https://github.com/Lightning-AI/pytorch-lightning/releases/tag/2.6.0
---
 pkgs/development/python-modules/pytorch-lightning/default.nix | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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

From e2268b7f8cc67a478793502b0dc62feb1aa897f3 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Sat, 29 Nov 2025 22:15:28 +0000
Subject: [PATCH 2/2] python3Packages.gluonts: fetch patch to fix failing test

https://github.com/awslabs/gluonts/pull/3269
---
 .../python-modules/gluonts/default.nix          | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

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"