python312Packages.trl: init at 0.15.2

{

  lib,

  buildPythonPackage,

  fetchFromGitHub,

  # build-system

  setuptools,

  setuptools-scm,

  # dependencies

  accelerate,

  datasets,

  rich,

  transformers,

}:

buildPythonPackage rec {

  pname = "trl";

  version = "0.15.2";

  pyproject = true;

  src = fetchFromGitHub {

    owner = "huggingface";

    repo = "trl";

    tag = "v${version}";

    hash = "sha256-HsSmFXFqDOWVLa6VXdPZVS9C3bjYcsliR0TwNpPiQx4=";

  };

  build-system = [

    setuptools

    setuptools-scm

  ];

  dependencies = [

    accelerate

    datasets

    rich

    transformers

  ];

  # Many tests require internet access.

  doCheck = false;

  pythonImportsCheck = [ "trl" ];

  meta = {

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

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

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

    license = lib.licenses.asl20;

    maintainers = with lib.maintainers; [ hoh ];

  };

}

  tritonclient = callPackage ../development/python-modules/tritonclient { };

  trl = callPackage ../development/python-modules/trl { };

  trlib = toPythonModule (

    pkgs.trlib.override {

      pythonSupport = true;