Add venvshellhook to jupyter

jupyter/flake.nix

@@ -3,26 +3,37 @@
 
   inputs.nixpkgs.url = "https://flakehub.com/f/NixOS/nixpkgs/0.1.*.tar.gz";
 
-  outputs =
+  outputs = {
-    { self
+    self,
-    , nixpkgs
+    nixpkgs,
-    ,
+  }: let
-    }:
+    supportedSystems = ["x86_64-linux" "aarch64-linux" "x86_64-darwin" "aarch64-darwin"];
-    let
+    forEachSupportedSystem = f:
-      supportedSystems = [ "x86_64-linux" "aarch64-linux" "x86_64-darwin" "aarch64-darwin" ];
+      nixpkgs.lib.genAttrs supportedSystems (system:
-      forEachSupportedSystem = f:
+        f {
-        nixpkgs.lib.genAttrs supportedSystems (system:
+          pkgs = import nixpkgs {inherit system;};
-          f {
+        });
-            pkgs = import nixpkgs { inherit system; };
+  in {
-          });
+    devShells = forEachSupportedSystem ({pkgs}: {
-    in
+      default = pkgs.mkShell {
-    {
+        packages = with pkgs;
-      devShells = forEachSupportedSystem ({ pkgs }: {
+          [python311 virtualenv]
-        default = pkgs.mkShell {
+          ++ (with pkgs.python311Packages; [
-          packages = with pkgs;
+            pip
-            [ python311 virtualenv ]
+            venvShellHook
-            ++ (with pkgs.python311Packages; [ pip requests jupyter pandas numpy matplotlib seaborn scikit-learn plotnine arrow polars pyarrow ]);
+            requests
-        };
+            jupyter
-      });
+            pandas
-    };
+            numpy
+            matplotlib
+            seaborn
+            scikit-learn
+            plotnine
+            arrow
+            polars
+            pyarrow
+          ]);
+      };
+    });
+  };
 }