pkgs/development/python-modules/docling-ibm-models/default.nix

   1 {
   2   lib,
   3   buildPythonPackage,
   4   fetchFromGitHub,
   5   poetry-core,
   6   huggingface-hub,
   7   jsonlines,
   8   lxml,
   9   mean-average-precision,
  10   numpy,
  11   opencv-python-headless,
  12   pillow,
  13   torch,
  14   torchvision,
  15   tqdm,
  16   pytestCheckHook,
  17 }:
  18
  19 buildPythonPackage rec {
  20   pname = "docling-ibm-models";
  21   version = "2.0.3";
  22   pyproject = true;
  23
  24   src = fetchFromGitHub {
  25     owner = "DS4SD";
  26     repo = "docling-ibm-models";
  27     rev = "refs/tags/v${version}";
  28     hash = "sha256-vSEW1+mFTjUvjjUOoX3aGgT/y8iwP3JGIZaPh9RbX5I=";
  29   };
  30
  31   build-system = [
  32     poetry-core
  33   ];
  34
  35   dependencies = [
  36     huggingface-hub
  37     jsonlines
  38     lxml
  39     mean-average-precision
  40     numpy
  41     opencv-python-headless
  42     pillow
  43     torch
  44     torchvision
  45     tqdm
  46   ];
  47
  48   pythonRelaxDeps = [
  49     "lxml"
  50     "mean_average_precision"
  51     "torchvision"
  52   ];
  53
  54   pythonImportsCheck = [
  55     "docling_ibm_models"
  56   ];
  57
  58   nativeCheckInputs = [
  59     pytestCheckHook
  60   ];
  61
  62   disabledTests = [
  63     # Requires network access
  64     "test_layoutpredictor"
  65     "test_tf_predictor"
  66   ];
  67
  68   meta = {
  69     changelog = "https://github.com/DS4SD/docling-ibm-models/blob/${src.rev}/CHANGELOG.md";
  70     description = "Docling IBM models";
  71     homepage = "https://github.com/DS4SD/docling-ibm-models";
  72     license = lib.licenses.mit;
  73     maintainers = with lib.maintainers; [ drupol ];
  74   };
  75 }