1 files changed, 25 insertions, 0 deletions
diff --git a/tasks/default.nix b/tasks/default.nix
index 3521443..1ccda33 100644
--- a/tasks/default.nix
+++ b/tasks/default.nix
@@ -27,6 +27,15 @@ let
     '';
   };
 
+  jupyter = import (builtins.fetchGit {
+    url = https://github.com/tweag/jupyterWith;
+    rev = "";
+  }) {};
+
+  mkJupyterEnv = kernel: (jupyter.jupyterlabWith {
+    kernels = [ kernel ];
+  }).env;
+
 in {
   custom.tasks = {
     bsc_edit = {
@@ -101,5 +110,21 @@ in {
         sympy.init_session()
       '';
     };
+
+    pymath_jupyter = {
+      description = "Python @ Jupyter Lab";
+      directory = "~/";
+      type = "jupyter-lab";
+      environment = mkJupyterEnv (
+        jupyter.kernels.iPythonWith {
+          name = "python";
+          packages = p: with p; [
+            numpy
+            sympy
+            matplotlib
+          ];
+        }
+      );
+    };
   };
 }