python3Packages.pytorch-lightning: 2.5.2 -> 2.5.3 (#433497)
This commit is contained in:
@@ -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 = ''
|
||||
|
||||
Reference in New Issue
Block a user