From 5fddb77c67e45d7d8af562f29acdd9b44f0c8449 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Mon, 23 Mar 2026 21:49:33 +0000
Subject: [PATCH] python3Packages.helion: 0.2.10 -> 0.3.2

Diff: https://github.com/pytorch/helion/compare/v0.2.10...v0.3.2

Changelog: https://github.com/pytorch/helion/releases/tag/v0.3.2
---
 pkgs/development/python-modules/helion/default.nix | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/pkgs/development/python-modules/helion/default.nix b/pkgs/development/python-modules/helion/default.nix
index cfd72e377006..2e26a9c30b78 100644
--- a/pkgs/development/python-modules/helion/default.nix
+++ b/pkgs/development/python-modules/helion/default.nix
@@ -12,6 +12,7 @@
   filecheck,
   psutil,
   rich,
+  scikit-learn,
   torch,
   tqdm,
   triton,
@@ -23,16 +24,16 @@
   helion,
 }:
 
-buildPythonPackage rec {
+buildPythonPackage (finalAttrs: {
   pname = "helion";
-  version = "0.2.10";
+  version = "0.3.2";
   pyproject = true;
 
   src = fetchFromGitHub {
     owner = "pytorch";
     repo = "helion";
-    tag = "v${version}";
-    hash = "sha256-kZyay9X2RcN3by+2oFAjt17Zuu34i3p+MeApBuhejmg=";
+    tag = "v${finalAttrs.version}";
+    hash = "sha256-MR8fo6MhMSAXlBsx//ODIprXPXFhMy6K5Mno9BnZGHc=";
   };
 
   build-system = [
@@ -44,6 +45,7 @@ buildPythonPackage rec {
     filecheck
     psutil
     rich
+    scikit-learn
     torch
     tqdm
     triton
@@ -75,10 +77,10 @@ buildPythonPackage rec {
   meta = {
     description = "Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate";
     homepage = "https://github.com/pytorch/helion";
-    changelog = "https://github.com/pytorch/helion/releases/tag/${src.tag}";
+    changelog = "https://github.com/pytorch/helion/releases/tag/${finalAttrs.src.tag}";
     license = lib.licenses.bsd3;
     maintainers = with lib.maintainers; [ GaetanLepage ];
     # This package explicitly requires CUDA-enabled pytorch
     broken = !config.cudaSupport;
   };
-}
+})