From 6666d4ae21f3b80c2542de754a5b0b3e2dd463ea Mon Sep 17 00:00:00 2001
From: Evie Litherland-Smith <evie.litherland-smith@ukaea.uk>
Date: Wed, 10 Jul 2024 17:13:01 +0100
Subject: [PATCH] Update pythonshell template

---
 templates | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/templates b/templates
index 685c865..d98695c 100644
--- a/templates
+++ b/templates
@@ -172,14 +172,13 @@ nix-mode
             )
 (pythonshell "let" n
              > "pkgs = import " (p "<nixpkgs>") " {};" n
-             > "in pkgs.mkShellNoCC {" n
-             > "packages = with pkgs; [" n
-             > "poetry" n
-             > (p "python3") n
-             > q n
-             > "];" n
+             > "poetry = pkgs.poetry;" n
+             > "python = pkgs." (p "python3") ";" n
+             > "in pkgs.mkShell {" n
+             > "packages = [ poetry " q "];" n
              > "shellHook = ''" n
              > "export LD_LIBRARY_PATH=${pkgs.stdenv.cc.cc.lib}/lib" n
+             > "${poetry}/bin/poetry env use ${python}/bin/python3" n
              > "'';" n
              > "}")