From e59c0bb38bfdfa60c033824876c62780287cda57 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Tue, 30 Jun 2026 22:42:44 +0000
Subject: [PATCH] python3Packages.torchsummary: cleanup, add missing
 dependencies

---
 .../python-modules/torchsummary/default.nix          | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/pkgs/development/python-modules/torchsummary/default.nix b/pkgs/development/python-modules/torchsummary/default.nix
index a3465370d5e2..7c0c4c583363 100644
--- a/pkgs/development/python-modules/torchsummary/default.nix
+++ b/pkgs/development/python-modules/torchsummary/default.nix
@@ -2,7 +2,12 @@
   lib,
   buildPythonPackage,
   fetchPypi,
+
+  # build-system
   setuptools,
+
+  # dependencies
+  numpy,
   torch,
 }:
 
@@ -10,7 +15,9 @@ buildPythonPackage (finalAttrs: {
   pname = "torchsummary";
   version = "1.5.1";
   pyproject = true;
+  __structuredAttrs = true;
 
+  # No tags on GitHub
   src = fetchPypi {
     inherit (finalAttrs) pname version;
     hash = "sha256-mBv2ieIuDPf5XHRgAvIKJK0mqmudhhE0oUvGzpIjBZA=";
@@ -18,7 +25,10 @@ buildPythonPackage (finalAttrs: {
 
   build-system = [ setuptools ];
 
-  dependencies = [ torch ];
+  dependencies = [
+    numpy
+    torch
+  ];
 
   # no tests in pypi tarball
   doCheck = false;