diff --git a/pkgs/development/python-modules/finetuning-scheduler/default.nix b/pkgs/development/python-modules/finetuning-scheduler/default.nix index 344e3c735659..5bef00a10bb1 100644 --- a/pkgs/development/python-modules/finetuning-scheduler/default.nix +++ b/pkgs/development/python-modules/finetuning-scheduler/default.nix @@ -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" ]; diff --git a/pkgs/development/python-modules/pytorch-lightning/default.nix b/pkgs/development/python-modules/pytorch-lightning/default.nix index 6a8d934b0c9d..d1c9b0fdd4c0 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.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 = ''