python312Packages.torch: cleanup dependencies

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