python3Packages.pytorch-lightning: 2.5.2 -> 2.5.3 (#433497)

This commit is contained in:
Gaétan Lepage
2025-08-14 11:12:50 +02:00
committed by GitHub
2 changed files with 24 additions and 19 deletions
@@ -58,23 +58,28 @@ buildPythonPackage rec {
nativeCheckInputs = [ pytestCheckHook ];
enabledTestPaths = [ "tests" ];
disabledTests =
lib.optionals (pythonOlder "3.12") [
# torch._dynamo.exc.BackendCompilerFailed: backend='inductor' raised:
# LoweringException: ImportError: cannot import name 'triton_key' from 'triton.compiler.compiler'
"test_fts_dynamo_enforce_p0"
"test_fts_dynamo_resume"
"test_fts_dynamo_intrafit"
]
++ lib.optionals (pythonAtLeast "3.13") [
# RuntimeError: Dynamo is not supported on Python 3.13+
"test_fts_dynamo_enforce_p0"
"test_fts_dynamo_resume"
]
++ lib.optionals (stdenv.hostPlatform.isAarch64 && stdenv.hostPlatform.isLinux) [
# slightly exceeds numerical tolerance on aarch64-linux:
"test_fts_frozen_bn_track_running_stats"
];
disabledTests = [
# Fails since pytorch-lightning was bumped to 2.5.3
# IndexError: list index out of range
# https://github.com/speediedan/finetuning-scheduler/issues/19
"test_fts_misconfiguration"
]
++ lib.optionals (pythonOlder "3.12") [
# torch._dynamo.exc.BackendCompilerFailed: backend='inductor' raised:
# LoweringException: ImportError: cannot import name 'triton_key' from 'triton.compiler.compiler'
"test_fts_dynamo_enforce_p0"
"test_fts_dynamo_resume"
"test_fts_dynamo_intrafit"
]
++ lib.optionals (pythonAtLeast "3.13") [
# RuntimeError: Dynamo is not supported on Python 3.13+
"test_fts_dynamo_enforce_p0"
"test_fts_dynamo_resume"
]
++ lib.optionals (stdenv.hostPlatform.isAarch64 && stdenv.hostPlatform.isLinux) [
# slightly exceeds numerical tolerance on aarch64-linux:
"test_fts_frozen_bn_track_running_stats"
];
pythonImportsCheck = [ "finetuning_scheduler" ];
@@ -24,14 +24,14 @@
buildPythonPackage rec {
pname = "pytorch-lightning";
version = "2.5.2";
version = "2.5.3";
pyproject = true;
src = fetchFromGitHub {
owner = "Lightning-AI";
repo = "pytorch-lightning";
tag = version;
hash = "sha256-zYPjMahC8/9uYomzSFMbswJoh+CcBsnO0h4pkgCxIcQ=";
hash = "sha256-xLm8t1A6mwIyX5WCkhdt9D9/PsomkSTCAxIGBQlFGxY=";
};
preConfigure = ''