about summary refs log tree commit diff
path: root/pkgs/development/python-modules/functorch/default.nix
diff options
context:
space:
mode:
Diffstat (limited to 'pkgs/development/python-modules/functorch/default.nix')
-rw-r--r--pkgs/development/python-modules/functorch/default.nix5
1 files changed, 5 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/functorch/default.nix b/pkgs/development/python-modules/functorch/default.nix
index 05b96077edc93..e7f4e29fe94a5 100644
--- 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/
   '';