add flake-jupyter

flake.nix

@@ -12,6 +12,10 @@
         path = ./flakes/rust;
         description = "Flake to develop, build and run a rust application";
       };
+      flake-jupyter = {
+        path = ./flakes/jupyter;
+        description = "Flake to configure jupyter lab";
+      };
       # starters
       starter-typescript = {
         path = ./starters/typescript;

flakes/jupyter/flake.nix

@@ -0,0 +1,33 @@
+{
+  inputs = {
+    jupyterWith.url = "github:tweag/jupyterWith";
+    utils.url = "github:numtide/flake-utils";
+  };
+
+  outputs = { self, nixpkgs, jupyterWith, utils }:
+    utils.lib.eachDefaultSystem (system:
+      let
+        pkgs = import nixpkgs {
+          inherit system;
+          overlays = builtins.attrValues jupyterWith.overlays;
+        };
+
+        # Source https://github.com/tweag/jupyterWith
+        iPython = pkgs.kernels.iPythonWith {
+          name = "python";
+          packages = p: with p; [ numpy ];
+        };
+
+        jupyterLab = pkgs.jupyterlabWith {
+          kernels = [ iPython ];
+        };
+
+      in
+      {
+        apps.default = {
+          type = "app";
+          program = "${jupyterLab}/bin/jupyter-lab";
+        };
+        devShell = jupyterLab.env;
+      });
+}