diff --git a/pkgs/development/python-modules/docling-ibm-models/default.nix b/pkgs/development/python-modules/docling-ibm-models/default.nix index 78665be7beab..7a91de39c52b 100644 --- a/pkgs/development/python-modules/docling-ibm-models/default.nix +++ b/pkgs/development/python-modules/docling-ibm-models/default.nix @@ -28,22 +28,27 @@ writableTmpDirAsHomeHook, }: -buildPythonPackage rec { +buildPythonPackage (finalAttrs: { pname = "docling-ibm-models"; - version = "3.11.0"; + version = "3.13.0"; pyproject = true; src = fetchFromGitHub { owner = "docling-project"; repo = "docling-ibm-models"; - tag = "v${version}"; - hash = "sha256-foRoxuTqwNqn2q/3pAXNoiUYrAKwzXVnAabNRietZ40="; + tag = "v${finalAttrs.version}"; + hash = "sha256-T8sVXG9s7jlhoRNexPRmCaiHPtQUAhDa9Z0Ri9i0zcc="; }; build-system = [ poetry-core ]; + pythonRelaxDeps = [ + "jsonlines" + "numpy" + "transformers" + ]; dependencies = [ accelerate docling-core @@ -61,15 +66,7 @@ buildPythonPackage rec { transformers ]; - pythonRelaxDeps = [ - "jsonlines" - "numpy" - "transformers" - ]; - - pythonImportsCheck = [ - "docling_ibm_models" - ]; + pythonImportsCheck = [ "docling_ibm_models" ]; nativeCheckInputs = [ datasets @@ -77,8 +74,13 @@ buildPythonPackage rec { writableTmpDirAsHomeHook ]; + disabledTestPaths = [ + # Require network access + "tests/test_tableformer_v2.py" + ]; + disabledTests = [ - # Requires network access + # Require network access "test_code_formula_predictor" # huggingface_hub.errors.LocalEntryNotFoundError "test_figure_classifier" # huggingface_hub.errors.LocalEntryNotFoundError "test_layoutpredictor" @@ -87,10 +89,10 @@ buildPythonPackage rec { ]; meta = { - changelog = "https://github.com/DS4SD/docling-ibm-models/blob/${src.tag}/CHANGELOG.md"; + changelog = "https://github.com/DS4SD/docling-ibm-models/blob/${finalAttrs.src.tag}/CHANGELOG.md"; description = "Docling IBM models"; homepage = "https://github.com/DS4SD/docling-ibm-models"; license = lib.licenses.mit; maintainers = [ ]; }; -} +})