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 = [ ]; }; -} +}) diff --git a/pkgs/development/python-modules/huggingface-hub/default.nix b/pkgs/development/python-modules/huggingface-hub/default.nix index f6721fd09a41..0d9992fe679c 100644 --- a/pkgs/development/python-modules/huggingface-hub/default.nix +++ b/pkgs/development/python-modules/huggingface-hub/default.nix @@ -37,14 +37,14 @@ buildPythonPackage (finalAttrs: { pname = "huggingface-hub"; - version = "1.9.0"; + version = "1.10.2"; pyproject = true; src = fetchFromGitHub { owner = "huggingface"; repo = "huggingface_hub"; tag = "v${finalAttrs.version}"; - hash = "sha256-7spvenDWuSDVTlcbOQd8oJ3bwzsyuQDIyMiVSyFIhuQ="; + hash = "sha256-Q9N0QnxV8oJcxUsJzv4wX8Z6FkNdEfUH5BEVoZolsRY="; }; build-system = [ setuptools ]; diff --git a/pkgs/development/python-modules/sentence-transformers/default.nix b/pkgs/development/python-modules/sentence-transformers/default.nix index 7983ee7d711f..743adf0efdcb 100644 --- a/pkgs/development/python-modules/sentence-transformers/default.nix +++ b/pkgs/development/python-modules/sentence-transformers/default.nix @@ -34,14 +34,14 @@ buildPythonPackage (finalAttrs: { pname = "sentence-transformers"; - version = "5.3.0"; + version = "5.4.1"; pyproject = true; src = fetchFromGitHub { owner = "huggingface"; repo = "sentence-transformers"; tag = "v${finalAttrs.version}"; - hash = "sha256-xf5ujZH7OH81ofavytI/Zd0PCkRf6rIoXzWI9kUjoDE="; + hash = "sha256-VZu50DVuU0P7o3+iKVWougui7nWSrnP/eza0Rqtt7ZU="; }; build-system = [ setuptools ]; @@ -58,9 +58,11 @@ buildPythonPackage (finalAttrs: { ]; optional-dependencies = { - image = [ - pillow - ]; + image = transformers.optional-dependencies.vision; + inherit (transformers.optional-dependencies) + audio + video + ; train = [ accelerate datasets @@ -115,12 +117,14 @@ buildPythonPackage (finalAttrs: { "test_router_with_trainer" "test_router_with_trainer_without_router_mapping" "test_save_and_load" + "test_simple" "test_simple_encode" "test_tokenize" "test_train_stsb" "test_trainer" "test_trainer_invalid_column_names" "test_trainer_multi_dataset_errors" + "test_unsupported_modality" "test_valid_initialization_no_warnings" "test_valid_initialization_with_weights" "test_weights_length_mismatch_raises_error" @@ -152,31 +156,26 @@ buildPythonPackage (finalAttrs: { disabledTestPaths = [ # Tests require network access - "tests/cross_encoder/test_cross_encoder.py" + "tests/base/modules/" + "tests/base/test_model.py" + "tests/base/test_model_card.py" + "tests/cross_encoder/evaluation/test_reranking.py" + "tests/cross_encoder/test_model.py" "tests/cross_encoder/test_model_card.py" "tests/cross_encoder/test_train_stsb.py" - "tests/evaluation/test_information_retrieval_evaluator.py" - "tests/sparse_encoder/models/test_csr.py" - "tests/sparse_encoder/models/test_sparse_static_embedding.py" + "tests/sentence_transformer/test_compute_embeddings.py" + "tests/sentence_transformer/test_model.py" + "tests/sentence_transformer/test_model_card.py" + "tests/sentence_transformer/test_model_card_data.py" + "tests/sparse_encoder/modules/test_csr.py" + "tests/sparse_encoder/modules/test_sparse_static_embedding.py" + "tests/sparse_encoder/test_model.py" "tests/sparse_encoder/test_model_card.py" "tests/sparse_encoder/test_opensearch_models.py" "tests/sparse_encoder/test_pretrained.py" - "tests/sparse_encoder/test_sparse_encoder.py" - "tests/test_compute_embeddings.py" - "tests/test_model_card.py" - "tests/test_model_card_data.py" - "tests/test_multi_process.py" - "tests/test_pretrained_stsb.py" - "tests/test_sentence_transformer.py" - "tests/test_train_stsb.py" "tests/util/test_hard_negatives.py" ]; - # Sentence-transformer needs a writable hf_home cache - postInstall = '' - export HF_HOME=$(mktemp -d) - ''; - meta = { description = "Multilingual Sentence & Image Embeddings with BERT"; homepage = "https://github.com/huggingface/sentence-transformers"; diff --git a/pkgs/development/python-modules/transformers/default.nix b/pkgs/development/python-modules/transformers/default.nix index 53d915d13f6a..f10f55689c3d 100644 --- a/pkgs/development/python-modules/transformers/default.nix +++ b/pkgs/development/python-modules/transformers/default.nix @@ -94,14 +94,14 @@ buildPythonPackage (finalAttrs: { pname = "transformers"; - version = "5.5.0"; + version = "5.5.4"; pyproject = true; src = fetchFromGitHub { owner = "huggingface"; repo = "transformers"; tag = "v${finalAttrs.version}"; - hash = "sha256-2fOCORAsQDKxp6EPe1OHysPWb/q168z6sCYg89tRXdU="; + hash = "sha256-ZqynYPj8VxH6BmvxHuw3lq16e2FFi3p8pw5of+vkz40="; }; build-system = [ setuptools ]; diff --git a/pkgs/development/python-modules/unsloth-zoo/default.nix b/pkgs/development/python-modules/unsloth-zoo/default.nix index 810628911f66..016b7b155ff3 100644 --- a/pkgs/development/python-modules/unsloth-zoo/default.nix +++ b/pkgs/development/python-modules/unsloth-zoo/default.nix @@ -39,14 +39,14 @@ buildPythonPackage (finalAttrs: { pname = "unsloth-zoo"; - version = "2026.4.2"; + version = "2026.4.7"; pyproject = true; # no tags on GitHub src = fetchPypi { pname = "unsloth_zoo"; inherit (finalAttrs) version; - hash = "sha256-l0OTaZjPrNnrxVYIfZcf6pYr1tJS9EGj+iguU6S+D28="; + hash = "sha256-jJ58d2+5lEALEaASELZtQkY2YxNWaLrfLvOCUGnwrh4="; }; postPatch = '' @@ -59,12 +59,10 @@ buildPythonPackage (finalAttrs: { "setuptools-scm" ''; - # Upstream constrains datasets/torch more tightly than the versions - # currently shipped in nixpkgs, but the package still builds and works with - # the newer dependency set here. pythonRelaxDeps = [ "datasets" "torch" + "transformers" ]; patches = [ diff --git a/pkgs/development/python-modules/unsloth/default.nix b/pkgs/development/python-modules/unsloth/default.nix index 07447392d21f..3f107752bb84 100644 --- a/pkgs/development/python-modules/unsloth/default.nix +++ b/pkgs/development/python-modules/unsloth/default.nix @@ -54,14 +54,14 @@ in buildPythonPackage (finalAttrs: { pname = "unsloth"; - version = "2026.4.1"; + version = "2026.4.5"; pyproject = true; # Tags on the GitHub repo don't match src = fetchPypi { pname = "unsloth"; inherit (finalAttrs) version; - hash = "sha256-RGpVxrcFzKbHxV7o0/OYaADo/Pqxx/c+FaxcV05gHGU="; + hash = "sha256-35+IMV/WHVi0iGnOxtfSZNKo0+0ZlNVlbNtA5tXw9sE="; }; postPatch = '' @@ -178,7 +178,6 @@ buildPythonPackage (finalAttrs: { meta = { description = "Finetune Llama 3.3, DeepSeek-R1 & Reasoning LLMs 2x faster with 70% less memory"; homepage = "https://github.com/unslothai/unsloth"; - changelog = "https://github.com/unslothai/unsloth/releases/tag/${finalAttrs.version}"; license = lib.licenses.asl20; maintainers = with lib.maintainers; [ hoh ]; };