Commit a424a303 authored by Hugo Herter's avatar Hugo Herter Committed by Gaetan Lepage
Browse files

python3Packages.trl: 0.27.0 -> 0.24.0 

https://github.com/huggingface/trl/releases/tag/v0.24.0

Keep trl within the version range required by unsloth and unsloth-zoo,
retain the missing packaging and rich runtime dependencies, and keep
the package expression on finalAttrs.
parent d3f7baaa

pkgs/development/python-modules/trl/default.nix

+9 −11
Original line number Diff line number Diff line
@@ -5,35 +5,33 @@ 


  # build-system

  setuptools,

  setuptools-scm,



  # dependencies

  accelerate,

  datasets,

  packaging,

  rich,

  transformers,

}:



buildPythonPackage rec {

buildPythonPackage (finalAttrs: {

  pname = "trl";

  version = "0.27.0";

  version = "0.24.0";

  pyproject = true;



  src = fetchFromGitHub {

    owner = "huggingface";

    repo = "trl";

    tag = "v${version}";

    hash = "sha256-NEvIWrirHqcLJpyA894NgNFPn/Svg+ND/xDMIRHW8d0=";

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

    hash = "sha256-t0wOuEKlcZzFlQeS4PYHykFsz+43hYc0gJ9u4emr8HI=";

  };



  build-system = [

    setuptools

    setuptools-scm

  ];

  build-system = [ setuptools ];



  dependencies = [

    accelerate

    datasets

    packaging

    rich

    transformers

  ];
@@ -46,8 +44,8 @@ buildPythonPackage rec { 
  meta = {

    description = "Train transformer language models with reinforcement learning";

    homepage = "https://github.com/huggingface/trl";

    changelog = "https://github.com/huggingface/trl/releases/tag/${src.tag}";

    changelog = "https://github.com/huggingface/trl/releases/tag/${finalAttrs.src.tag}";

    license = lib.licenses.asl20;

    maintainers = with lib.maintainers; [ hoh ];

  };

}
 
})