From b69e5f65db2a5b2d2ccd8833a4caa598c74e4c12 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Mon, 6 Jul 2026 15:26:23 +0000
Subject: [PATCH 1/2] python3Packages.torchtitan: add missing test dependency
 expecttest

---
 pkgs/development/python-modules/torchtitan/default.nix | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/pkgs/development/python-modules/torchtitan/default.nix b/pkgs/development/python-modules/torchtitan/default.nix
index e7e16da4040f..da624b457808 100644
--- a/pkgs/development/python-modules/torchtitan/default.nix
+++ b/pkgs/development/python-modules/torchtitan/default.nix
@@ -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

From a386bfeb1819c505891fbdbc32e6d8e4aac06ed9 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Mon, 6 Jul 2026 15:27:13 +0000
Subject: [PATCH 2/2] python3Packages.torchtitan: enable __structuredAttrs

---
 pkgs/development/python-modules/torchtitan/default.nix | 1 +
 1 file changed, 1 insertion(+)

diff --git a/pkgs/development/python-modules/torchtitan/default.nix b/pkgs/development/python-modules/torchtitan/default.nix
index da624b457808..3bf24a97c2bd 100644
--- a/pkgs/development/python-modules/torchtitan/default.nix
+++ b/pkgs/development/python-modules/torchtitan/default.nix
@@ -31,6 +31,7 @@ buildPythonPackage (finalAttrs: {
   pname = "torchtitan";
   version = "0.2.2";
   pyproject = true;
+  __structuredAttrs = true;
 
   src = fetchFromGitHub {
     owner = "pytorch";