python3Packages.torchtune: run only the torchtune tests

regression and recipies tests skipped.
2025-12-24 11:16:40 -08:00
parent 4ea3b99591
commit 31b3f2a72c
1 changed files with 3 additions and 0 deletions
@@ -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"