Unverified Commit ab5de2be authored by Gaétan Lepage's avatar Gaétan Lepage Committed by GitHub
Browse files

python3Packages.finetuning-scheduler: 2.9.2 -> 2.10.0 (#484060)

parents 47735b2a f9b4702a

pkgs/development/python-modules/finetuning-scheduler/default.nix

+12 −6
Original line number Diff line number Diff line
@@ -13,18 +13,19 @@ 


  # tests

  pytestCheckHook,

  pythonAtLeast,

}:



buildPythonPackage rec {

buildPythonPackage (finalAttrs: {

  pname = "finetuning-scheduler";

  version = "2.9.2";

  version = "2.10.0";

  pyproject = true;



  src = fetchFromGitHub {

    owner = "speediedan";

    repo = "finetuning-scheduler";

    tag = "v${version}";

    hash = "sha256-+LJ36LzFamC5Mv5ec+uUtMWZt0WMjuHKZlA73rZqoxw=";

    tag = "v${finalAttrs.version}";

    hash = "sha256-O8M0CW71DLDfpL35LbVZjtjOB+pdNvGOAYD4miM8jtM=";

  };



  build-system = [ setuptools ];
@@ -47,6 +48,11 @@ buildPythonPackage rec { 
    # AssertionError: assert 'lightning @ git+' in 'lightning>=2.5.0,<2.5.6'

    "test_get_lightning_requirement"

  ]

  ++ lib.optionals (pythonAtLeast "3.14") [

    # RuntimeError: torch.compile is not supported on Python 3.14+

    "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"
@@ -59,8 +65,8 @@ buildPythonPackage rec { 
  meta = {

    description = "PyTorch Lightning extension for foundation model experimentation with flexible fine-tuning schedules";

    homepage = "https://finetuning-scheduler.readthedocs.io";

    changelog = "https://github.com/speediedan/finetuning-scheduler/blob/v${version}/CHANGELOG.md";

    changelog = "https://github.com/speediedan/finetuning-scheduler/blob/${finalAttrs.src.tag}/CHANGELOG.md";

    license = lib.licenses.asl20;

    maintainers = with lib.maintainers; [ bcdarwin ];

  };

}
 
})