Commit 7417abd4 authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files
parent b8ba3ddc
Loading
Loading
Loading
Loading
+7 −9
Original line number Diff line number Diff line
@@ -22,21 +22,19 @@
  pytestCheckHook,
}:

buildPythonPackage rec {
buildPythonPackage (finalAttrs: {
  pname = "pytorch-lightning";
  version = "2.6.0";
  version = "2.6.1";
  pyproject = true;

  src = fetchFromGitHub {
    owner = "Lightning-AI";
    repo = "pytorch-lightning";
    tag = version;
    hash = "sha256-zmFA9/tz0C06LmQ37wHeoR1kBHKdoz/D1cKWMoeWHzs=";
    tag = finalAttrs.version;
    hash = "sha256-zOSV2X3yZy0uh1lJ2yNl/hHBvfIDcIrATHtiRwThsQA=";
  };

  preConfigure = ''
    export PACKAGE_NAME=pytorch
  '';
  env.PACKAGE_NAME = "pytorch";

  build-system = [ setuptools ];

@@ -67,8 +65,8 @@ buildPythonPackage rec {
  meta = {
    description = "Lightweight PyTorch wrapper for machine learning researchers";
    homepage = "https://github.com/Lightning-AI/pytorch-lightning";
    changelog = "https://github.com/Lightning-AI/pytorch-lightning/releases/tag/${src.tag}";
    changelog = "https://github.com/Lightning-AI/pytorch-lightning/releases/tag/${finalAttrs.src.tag}";
    license = lib.licenses.asl20;
    maintainers = with lib.maintainers; [ tbenst ];
  };
}
})