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" ];