Unverified Commit e43cf7b2 authored by kirillrdy's avatar kirillrdy Committed by GitHub
Browse files

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

parents a6f8ebd4 e2268b7f

pkgs/development/python-modules/gluonts/default.nix

+15 −2
Original line number Diff line number Diff line
@@ -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

+2 −2
Original line number Diff line number Diff line
@@ -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 = ''