Commit 44e8bf6e authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files

python312Packages.docling-ibm-models: 3.3.2 -> 3.4.1 

Diff: https://github.com/DS4SD/docling-ibm-models/compare/refs/tags/v3.3.2...v3.4.1

Changelog: https://github.com/DS4SD/docling-ibm-models/blob/v3.4.1/CHANGELOG.md
parent 05452072

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

+29 −21
Original line number Diff line number Diff line
@@ -2,31 +2,40 @@ 
  lib,

  buildPythonPackage,

  fetchFromGitHub,



  # build-system

  poetry-core,



  # dependencies

  torch,

  torchvision,

  transformers,

  docling-core,

  huggingface-hub,

  jsonlines,

  numpy,

  opencv-python-headless,

  pillow,

  tqdm,

  pydantic,

  safetensors,

  torch,

  torchvision,

  tqdm,

  transformers,



  # tests

  datasets,

  pytestCheckHook,

  writableTmpDirAsHomeHook,

}:



buildPythonPackage rec {

  pname = "docling-ibm-models";

  version = "3.3.2";

  version = "3.4.1";

  pyproject = true;



  src = fetchFromGitHub {

    owner = "DS4SD";

    repo = "docling-ibm-models";

    tag = "v${version}";

    hash = "sha256-8mqDgbTj5g6jXEumj16Me9NjHLCOdR+pXmAwn2dghfg=";

    hash = "sha256-IgKP+4en0N6xLBjzwmddDk8z5P3J1vo+UMW+65a01Ww=";

  };



  build-system = [
@@ -34,21 +43,21 @@ buildPythonPackage rec { 
  ];



  dependencies = [

    torch

    torchvision

    transformers

    numpy

    docling-core

    huggingface-hub

    jsonlines

    pillow

    tqdm

    numpy

    opencv-python-headless

    huggingface-hub

    pillow

    pydantic

    safetensors

    torch

    torchvision

    tqdm

    transformers

  ];



  pythonRelaxDeps = [

    "pillow"

    "torchvision"

    "transformers"

    "numpy"

  ];
@@ -58,19 +67,18 @@ buildPythonPackage rec { 
  ];



  nativeCheckInputs = [

    datasets

    pytestCheckHook

    writableTmpDirAsHomeHook

  ];



  preCheck = ''

    export HOME="$TEMPDIR"

  '';



  disabledTests = [

    # Requires network access

    "test_layoutpredictor"

    "test_tf_predictor"

    "test_code_formula_predictor" # huggingface_hub.errors.LocalEntryNotFoundError

    "test_figure_classifier" # huggingface_hub.errors.LocalEntryNotFoundError

    "test_layoutpredictor"

    "test_readingorder"

    "test_tf_predictor"

  ];



  meta = {