diff --git a/pkgs/development/python-modules/docling-ibm-models/default.nix b/pkgs/development/python-modules/docling-ibm-models/default.nix index 38137792b16c..069d29cb6a0e 100644 --- a/pkgs/development/python-modules/docling-ibm-models/default.nix +++ b/pkgs/development/python-modules/docling-ibm-models/default.nix @@ -1,5 +1,6 @@ { lib, + stdenv, buildPythonPackage, fetchFromGitHub, @@ -58,8 +59,9 @@ buildPythonPackage rec { ]; pythonRelaxDeps = [ - "transformers" + "jsonlines" "numpy" + "transformers" ]; pythonImportsCheck = [ @@ -81,6 +83,12 @@ buildPythonPackage rec { "test_tf_predictor" ]; + disabledTestPaths = lib.optionals stdenv.hostPlatform.isDarwin [ + # No module named 'torch._C._distributed_c10d'; 'torch._C' is not a package + "tests/test_code_formula_predictor.py" + "tests/test_layout_predictor.py" + ]; + meta = { changelog = "https://github.com/DS4SD/docling-ibm-models/blob/${src.tag}/CHANGELOG.md"; description = "Docling IBM models"; diff --git a/pkgs/development/python-modules/docling/default.nix b/pkgs/development/python-modules/docling/default.nix index 21df13d97842..9e792da7e8de 100644 --- a/pkgs/development/python-modules/docling/default.nix +++ b/pkgs/development/python-modules/docling/default.nix @@ -1,5 +1,6 @@ { lib, + stdenv, buildPythonPackage, fetchFromGitHub, @@ -160,6 +161,25 @@ buildPythonPackage rec { "test_e2e_valid_csv_conversions" ]; + disabledTestPaths = lib.optionals stdenv.hostPlatform.isDarwin [ + # No module named 'torch._C._distributed_c10d'; 'torch._C' is not a package + "tests/test_backend_csv.py" + "tests/test_backend_html.py" + "tests/test_backend_jats.py" + "tests/test_backend_msexcel.py" + "tests/test_backend_msword.py" + "tests/test_backend_pptx.py" + "tests/test_cli.py" + "tests/test_code_formula.py" + "tests/test_document_picture_classifier.py" + "tests/test_e2e_conversion.py" + "tests/test_e2e_ocr_conversion.py" + "tests/test_interfaces.py" + "tests/test_invalid_input.py" + "tests/test_legacy_format_transform.py" + "tests/test_options.py" + ]; + meta = { description = "Get your documents ready for gen AI"; homepage = "https://github.com/DS4SD/docling";