python312Packages.docling-ibm-models: init at 2.0.3

{

  lib,

  buildPythonPackage,

  fetchFromGitHub,

  poetry-core,

  huggingface-hub,

  jsonlines,

  lxml,

  mean-average-precision,

  numpy,

  opencv-python-headless,

  pillow,

  torch,

  torchvision,

  tqdm,

  pytestCheckHook,

}:

buildPythonPackage rec {

  pname = "docling-ibm-models";

  version = "2.0.3";

  pyproject = true;

  src = fetchFromGitHub {

    owner = "DS4SD";

    repo = "docling-ibm-models";

    rev = "refs/tags/v${version}";

    hash = "sha256-vSEW1+mFTjUvjjUOoX3aGgT/y8iwP3JGIZaPh9RbX5I=";

  };

  build-system = [

    poetry-core

  ];

  dependencies = [

    huggingface-hub

    jsonlines

    lxml

    mean-average-precision

    numpy

    opencv-python-headless

    pillow

    torch

    torchvision

    tqdm

  ];

  pythonRelaxDeps = [

    "lxml"

    "mean_average_precision"

    "torchvision"

  ];

  pythonImportsCheck = [

    "docling_ibm_models"

  ];

  nativeCheckInputs = [

    pytestCheckHook

  ];

  disabledTests = [

    # Requires network access

    "test_layoutpredictor"

    "test_tf_predictor"

  ];

  meta = {

    changelog = "https://github.com/DS4SD/docling-ibm-models/blob/${src.rev}/CHANGELOG.md";

    description = "Docling IBM models";

    homepage = "https://github.com/DS4SD/docling-ibm-models";

    license = lib.licenses.mit;

    maintainers = with lib.maintainers; [ drupol ];

  };

}

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

  docling-ibm-models = callPackage ../development/python-modules/docling-ibm-models { };

  docling-parse = callPackage ../development/python-modules/docling-parse {

    loguru-cpp = pkgs.loguru;

  };