python3Packages.functorch: add pybind11 as dependency

2022-06-15 11:15:37 -07:00 · 2022-06-15 11:15:37 -07:00 · 1f94955861
commit 1f94955861
parent 605ce577fa
1 changed files with 5 additions and 0 deletions
--- a/pkgs/development/python-modules/functorch/default.nix
+++ b/pkgs/development/python-modules/functorch/default.nix
@ -6,6 +6,7 @@
 , pytestCheckHook
 , python
 , pytorch
+, pybind11
 , which
 }:

@ -29,6 +30,10 @@ buildPythonPackage rec {
    which
  ];

+  buildInputs = [
+    pybind11
+  ];
+
  preCheck = ''
    rm -rf functorch/
  '';