From adf584d0443a1b7cec4be80db890edd146a03286 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Mon, 6 Apr 2026 20:42:51 +0000
Subject: [PATCH] python3Packages.helion: 0.3.2 -> 1.0.0

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

Changelog: https://github.com/pytorch/helion/releases/tag/v1.0.0
---
 .../python-modules/helion/default.nix         | 25 +++++++++----------
 1 file changed, 12 insertions(+), 13 deletions(-)

diff --git a/pkgs/development/python-modules/helion/default.nix b/pkgs/development/python-modules/helion/default.nix
index 2e26a9c30b78..9da96d721d41 100644
--- a/pkgs/development/python-modules/helion/default.nix
+++ b/pkgs/development/python-modules/helion/default.nix
@@ -10,13 +10,12 @@
 
   # dependencies
   filecheck,
+  numpy,
   psutil,
   rich,
   scikit-learn,
-  torch,
-  tqdm,
-  triton,
   typing-extensions,
+  torch,
 
   # tests
   pytestCheckHook,
@@ -26,14 +25,14 @@
 
 buildPythonPackage (finalAttrs: {
   pname = "helion";
-  version = "0.3.2";
+  version = "1.0.0";
   pyproject = true;
 
   src = fetchFromGitHub {
     owner = "pytorch";
     repo = "helion";
     tag = "v${finalAttrs.version}";
-    hash = "sha256-MR8fo6MhMSAXlBsx//ODIprXPXFhMy6K5Mno9BnZGHc=";
+    hash = "sha256-3Iam+1FUg9I9kKmkQBZp9/FTZpEjf4Ba+cKRo5eLEzw=";
   };
 
   build-system = [
@@ -43,13 +42,15 @@ buildPythonPackage (finalAttrs: {
 
   dependencies = [
     filecheck
+    numpy
     psutil
     rich
     scikit-learn
-    torch
-    tqdm
-    triton
     typing-extensions
+
+    # torch is not listed as a dependency, but is actually required at import time
+    # https://github.com/pytorch/helion/blob/v1.0.0/helion/_compat.py#L13
+    torch
   ];
 
   pythonImportsCheck = [ "helion" ];
@@ -67,11 +68,9 @@ buildPythonPackage (finalAttrs: {
 
   # Tests require GPU access
   doCheck = false;
-  passthru.gpuChecks = {
-    pytest = helion.overridePythonAttrs {
-      doCheck = true;
-      requiredSystemFeatures = [ "cuda" ];
-    };
+  passthru.gpuCheck = helion.overridePythonAttrs {
+    doCheck = true;
+    requiredSystemFeatures = [ "cuda" ];
   };
 
   meta = {