From d0b8de200fb219e803079f85a305a9d09e8fb3a0 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Sat, 22 Mar 2025 15:45:35 +0100
Subject: [PATCH] python312Packages.torch: cleanup dependencies

---
 .../python-modules/torch/default.nix          | 39 ++++++++++---------
 1 file changed, 20 insertions(+), 19 deletions(-)

diff --git a/pkgs/development/python-modules/torch/default.nix b/pkgs/development/python-modules/torch/default.nix
index c355e6e99e2a..2a3e54cab71b 100644
--- a/pkgs/development/python-modules/torch/default.nix
+++ b/pkgs/development/python-modules/torch/default.nix
@@ -44,15 +44,18 @@
 
   # dependencies
   astunparse,
-  fsspec,
+  expecttest,
   filelock,
+  fsspec,
+  hypothesis,
   jinja2,
   networkx,
-  sympy,
-  numpy,
+  packaging,
+  psutil,
   pyyaml,
-  cffi,
-  click,
+  requests,
+  sympy,
+  types-dataclasses,
   typing-extensions,
   # ROCm build and `torch.compile` requires `triton`
   tritonSupport ? (!stdenv.hostPlatform.isDarwin),
@@ -69,10 +72,6 @@
   _tritonEffective ? if cudaSupport then triton-cuda else triton,
   triton-cuda,
 
-  # Unit tests
-  hypothesis,
-  psutil,
-
   # Disable MKLDNN on aarch64-darwin, it negatively impacts performance,
   # this is also what official pytorch build does
   mklDnnSupport ? !(stdenv.hostPlatform.isDarwin && stdenv.hostPlatform.isAarch64),
@@ -510,18 +509,20 @@ buildPythonPackage rec {
   dependencies =
     [
       astunparse
-      cffi
-      click
-      numpy
-      pyyaml
-
-      # From install_requires:
-      fsspec
+      expecttest
       filelock
-      typing-extensions
-      sympy
-      networkx
+      fsspec
+      hypothesis
       jinja2
+      networkx
+      ninja
+      packaging
+      psutil
+      pyyaml
+      requests
+      sympy
+      types-dataclasses
+      typing-extensions
 
       # the following are required for tensorboard support
       pillow