python3Packages.torchtitan: add missing test dependency expecttest

2026-07-06 15:45:36 +00:00
parent 63016669ad
commit b69e5f65db
1 changed files with 2 additions and 0 deletions
@@ -20,6 +20,7 @@
  tyro,

  # tests
+  expecttest,
  pytestCheckHook,
  tomli-w,
  triton,
@@ -58,6 +59,7 @@ buildPythonPackage (finalAttrs: {
  pythonImportsCheck = [ "torchtitan" ];

  nativeCheckInputs = [
+    expecttest
    pytestCheckHook
    tomli-w
    transformers