Run jupyter automatically in python datasci

python-data-science.nix

@@ -1,8 +1,8 @@
-{ config, lib, pkgs, ... }:
+{ pkgs ? import <nixpkgs> { } }:
 
 with pkgs;
 
-{
+mkShell {
   buildInputs = [
     python3
     python3Packages.jupyter
@@ -12,4 +12,10 @@ with pkgs;
     python3Packages.matplotlib
     python3Packages.seaborn
   ];
+
+  shellHook = ''
+    trap "kill 0" EXIT
+
+    jupyter notebook
+  '';
 }