diff --git a/pkgs/development/python-modules/cut-cross-entropy/default.nix b/pkgs/development/python-modules/cut-cross-entropy/default.nix index dd9b90b42a64..6b7ecf81b715 100644 --- a/pkgs/development/python-modules/cut-cross-entropy/default.nix +++ b/pkgs/development/python-modules/cut-cross-entropy/default.nix @@ -24,7 +24,7 @@ buildPythonPackage { pname = "cut-cross-entropy"; - version = "25.5.1"; + version = "25.7.2"; pyproject = true; # The `ml-cross-entropy` Pypi comes from a third-party. @@ -32,8 +32,8 @@ buildPythonPackage { src = fetchFromGitHub { owner = "apple"; repo = "ml-cross-entropy"; - rev = "b616b222976b235647790a16d0388338b9e18941"; # no tags - hash = "sha256-BVPon+T7chkpozX/IZU3KZMw1zRzlYVvF/22JWKjT2Y="; + rev = "b19a424ed30a05b8261cfa84d83b2601a9454c67"; # no tags + hash = "sha256-AwUqKiI7XjEOZ7ofjQCOsqvxHyTFD4RZ70odPyxxntc="; }; build-system = [ @@ -65,6 +65,10 @@ buildPythonPackage { nativeCheckInputs = [ pytestCheckHook ]; + disabledTests = [ + "test_vocab_parallel" # Requires CUDA but does not use pytest.skip + ]; + pythonImportsCheck = [ "cut_cross_entropy" ]; diff --git a/pkgs/development/python-modules/trl/default.nix b/pkgs/development/python-modules/trl/default.nix index bc89ac18fd4b..e2b0b144b4c1 100644 --- a/pkgs/development/python-modules/trl/default.nix +++ b/pkgs/development/python-modules/trl/default.nix @@ -16,14 +16,14 @@ buildPythonPackage rec { pname = "trl"; - version = "0.20.0"; + version = "0.21.0"; pyproject = true; src = fetchFromGitHub { owner = "huggingface"; repo = "trl"; tag = "v${version}"; - hash = "sha256-z14refdNySnKcfFq54l+slsi4SLe5FG8UNoAKxfmAy0="; + hash = "sha256-9jbbbiGa/2NqHKe9rxDRyzfaWyy7tsoeHaMlpg0Oxk0="; }; build-system = [ diff --git a/pkgs/development/python-modules/unsloth/default.nix b/pkgs/development/python-modules/unsloth/default.nix index dc45bf8ded2f..73f94721b5e0 100644 --- a/pkgs/development/python-modules/unsloth/default.nix +++ b/pkgs/development/python-modules/unsloth/default.nix @@ -31,14 +31,14 @@ buildPythonPackage rec { pname = "unsloth"; - version = "2025.8.1"; + version = "2025.8.9"; pyproject = true; # Tags on the GitHub repo don't match src = fetchPypi { pname = "unsloth"; inherit version; - hash = "sha256-hkE+f6apgilrE0lFTWRe8PEvRQ71YuoHpQgzd/R/FaI="; + hash = "sha256-XKYKEJCiOucuGfout8+FiTfO3KXvkucaduwWnZR5OwA="; }; build-system = [ @@ -72,6 +72,7 @@ buildPythonPackage rec { # but it is not used. # Upstream issue: https://github.com/unslothai/unsloth-zoo/pull/68 pythonRelaxDeps = [ + "datasets" "protobuf" "transformers" "torch"