Commit 7932cca2 authored by happysalada's avatar happysalada
Browse files

python3Packages.docling-ibm-models: 2.0.4 -> 3.3.0 

docling-ibm-models: fix read onlyy file system

docling-ibm-models: disable broken tests
parent 2150d69a

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

+22 −12
Original line number Diff line number Diff line
@@ -3,28 +3,30 @@ 
  buildPythonPackage,

  fetchFromGitHub,

  poetry-core,

  # dependencies

  torch,

  torchvision,

  transformers,

  huggingface-hub,

  jsonlines,

  mean-average-precision,

  numpy,

  opencv-python-headless,

  pillow,

  torch,

  torchvision,

  tqdm,

  safetensors,

  pytestCheckHook,

}:



buildPythonPackage rec {

  pname = "docling-ibm-models";

  version = "2.0.4";

  version = "3.3.0";

  pyproject = true;



  src = fetchFromGitHub {

    owner = "DS4SD";

    repo = "docling-ibm-models";

    tag = "v${version}";

    hash = "sha256-QZvkkazxgkGuSQKIYI+YghH7pLlDSEbCGhg89gZsOpk=";

    hash = "sha256-wxkHd+TCBibOTWO09JOsjX6oBtUxZ/9IOmyLdeptzeQ=";

  };



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



  dependencies = [

    huggingface-hub

    jsonlines

    mean-average-precision

    numpy

    opencv-python-headless

    pillow

    torch

    torchvision

    transformers

    numpy

    jsonlines

    pillow

    tqdm

    opencv-python-headless

    huggingface-hub

    safetensors

  ];



  pythonRelaxDeps = [

    "mean_average_precision"

    "pillow"

    "torchvision"

    "transformers"

    "numpy"

  ];



  pythonImportsCheck = [
@@ -57,10 +61,16 @@ buildPythonPackage rec { 
    pytestCheckHook

  ];



  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

  ];



  meta = {