diff --git a/nix/impure-shell.nix b/nix/impure-shell.nix
index 92e14bc3..a13fd711 100644
--- a/nix/impure-shell.nix
+++ b/nix/impure-shell.nix
@@ -9,6 +9,7 @@
   cudaPackages,
   openssl,
   pkg-config,
+  poetry,
   protobuf,
   python3,
   pyright,
@@ -28,6 +29,7 @@ mkShell {
       black
       isort
       pkg-config
+      poetry
       (rust-bin.stable.latest.default.override {
         extensions = [
           "rust-analyzer"