From 32aad6a0c07602cd06347006e9bb70168b976f3c Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Wed, 6 May 2026 15:24:44 +0000
Subject: [PATCH] python3Packages.tt-flash: cleanup, enable tests

---
 .../python-modules/tt-flash/default.nix       | 20 +++++++++++++++----
 1 file changed, 16 insertions(+), 4 deletions(-)

diff --git a/pkgs/development/python-modules/tt-flash/default.nix b/pkgs/development/python-modules/tt-flash/default.nix
index 429f91f52a91..813e25fd0289 100644
--- a/pkgs/development/python-modules/tt-flash/default.nix
+++ b/pkgs/development/python-modules/tt-flash/default.nix
@@ -2,12 +2,18 @@
   lib,
   buildPythonPackage,
   fetchFromGitHub,
-  fetchpatch,
+
+  # build-system
   setuptools,
+
+  # dependencies
   pyyaml,
   tabulate,
   pyluwen,
   tt-tools-common,
+
+  # tests
+  pytestCheckHook,
 }:
 buildPythonPackage (finalAttrs: {
   pname = "tt-flash";
@@ -26,6 +32,10 @@ buildPythonPackage (finalAttrs: {
     setuptools
   ];
 
+  pythonRelaxDeps = [
+    "pyyaml"
+    "tabulate"
+  ];
   dependencies = [
     tabulate
     pyyaml
@@ -34,14 +44,16 @@ buildPythonPackage (finalAttrs: {
   ];
 
   pythonImportsCheck = [ "tt_flash" ];
-  pythonRelaxDeps = [
-    "pyyaml"
-    "tabulate"
+
+  nativeCheckInputs = [
+    pytestCheckHook
   ];
 
   meta = {
     description = "Tenstorrent Firmware Update Utility";
     homepage = "https://tenstorrent.com";
+    downloadPage = "https://github.com/tenstorrent/tt-flash";
+    changelog = "https://github.com/tenstorrent/tt-flash/blob/${finalAttrs.src.tag}/CHANGELOG.md";
     maintainers = with lib.maintainers; [ RossComputerGuy ];
     license = with lib.licenses; [ asl20 ];
   };