python312Packages.sentence-transformers: add optional-dependencies

  lib,

  buildPythonPackage,

  fetchFromGitHub,

  pythonOlder,

  # build-system

  setuptools,

  # dependencies

  accelerate,

  datasets,

  huggingface-hub,

  nltk,

  numpy,

  optimum,

  pillow,

  scikit-learn,

  scipy,

  sentencepiece,

  transformers,

  # tests

  accelerate,

  datasets,

  pytestCheckHook,

  pytest-cov-stub,

}:

  version = "3.4.1";

  pyproject = true;

  disabled = pythonOlder "3.9";

  src = fetchFromGitHub {

    owner = "UKPLab";

    repo = "sentence-transformers";

  dependencies = [

    huggingface-hub

    nltk

    numpy

    pillow

    scikit-learn

    scipy

    sentencepiece

    tokenizers

    torch

    tqdm

    transformers

  ];

  nativeCheckInputs = [

  optional-dependencies = {

    train = [

      accelerate

      datasets

    pytestCheckHook

    pytest-cov-stub

    ];

    onnx = [ optimum ] ++ optimum.optional-dependencies.onnxruntime;

    # onnx-gpu = [ optimum ] ++ optimum.optional-dependencies.onnxruntime-gpu;

    # openvino = [ optimum-intel ] ++ optimum-intel.optional-dependencies.openvino;

  };

  nativeCheckInputs = [

    pytest-cov-stub

    pytestCheckHook

  ] ++ lib.flatten (builtins.attrValues optional-dependencies);

  pythonImportsCheck = [ "sentence_transformers" ];