python310Packages.trainer: add missing input

2023-01-22 09:29:18 +01:00 · 2023-01-22 09:29:18 +01:00 · ee251dca0e
commit ee251dca0e
parent 831ce5cb71
1 changed files with 4 additions and 2 deletions
--- a/pkgs/development/python-modules/trainer/default.nix
+++ b/pkgs/development/python-modules/trainer/default.nix
@ -9,6 +9,7 @@
 , torch-bin
 , tensorboardx
 , protobuf
+, psutil

 , pytestCheckHook
 , soundfile
@ -37,10 +38,11 @@ buildPythonPackage {
  propagatedBuildInputs = [
    coqpit
    fsspec
-    torch-bin
+    protobuf
+    psutil
    soundfile
    tensorboardx
-    protobuf
+    torch-bin
  ];

  # only one test and that requires training data from the internet