python3Packages.torchtitan: make test dependency on triton explicit

2026-06-03 08:51:40 +00:00
parent c8560a2e8a
commit 6845478ade
1 changed files with 2 additions and 0 deletions
@@ -22,6 +22,7 @@
  # tests
  pytestCheckHook,
  tomli-w,
+  triton,
  writableTmpDirAsHomeHook,
 }:

@@ -60,6 +61,7 @@ buildPythonPackage (finalAttrs: {
    pytestCheckHook
    tomli-w
    transformers
+    triton
    writableTmpDirAsHomeHook
  ];