diff --git a/pkgs/development/python-modules/torchtune/default.nix b/pkgs/development/python-modules/torchtune/default.nix index d205aa25f51d..df55b169f30c 100644 --- a/pkgs/development/python-modules/torchtune/default.nix +++ b/pkgs/development/python-modules/torchtune/default.nix @@ -85,6 +85,9 @@ buildPythonPackage rec { writableTmpDirAsHomeHook ]; + # Exclude `regression` which depends on a specific llama model and `recipies` which are sample code + enabledTestPaths = [ "tests/torchtune" ]; + disabledTests = [ # AssertionError (tensors are not equal) "test_stop_tokens"