From 9adfae87ee64df1eb22549e82e7a5c79ed1e348e Mon Sep 17 00:00:00 2001
From: David Crompton <david.crompton@mail.utoronto.ca>
Date: Tue, 16 Jan 2024 15:00:48 -0500
Subject: [PATCH] Jupyterhub: QOL & NEST Fix (MPI remove)

---
 machines/kcnhub/servers/jupyterhub.nix | 21 ++++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

diff --git a/machines/kcnhub/servers/jupyterhub.nix b/machines/kcnhub/servers/jupyterhub.nix
index 9b2a10c..a9b97eb 100644
--- a/machines/kcnhub/servers/jupyterhub.nix
+++ b/machines/kcnhub/servers/jupyterhub.nix
@@ -8,29 +8,48 @@
       c.SystemdSpawner.mem_limit = '2G'
       c.SystemdSpawner.cpu_limit = 1.0
     '';
+
     kernels = {
       python3 = let
         env = (pkgs.python3.withPackages (pythonPackages: with pythonPackages; [
           # Necessary for use as a kernel
           ipykernel
+          ipdb
+          # Interactive widgets
+          ipywidgets
+          # Show images/media
+          mediapy
+          # Progress bars etc. for in Jupyter/IPython
+          halo
 
           # Common Python Libraries
           numpy
           scipy
           scikit-learn
           pandas
+          # Tables displayed in Jupyter: like Pandas Dataframes
+          ipytablewidgets
           matplotlib
+          ipympl
+          matplotlib-inline
           seaborn
 
           # Neural Data Processing Libraries
           # Temporarily broken?: https://github.com/NixOS/nixpkgs/issues/259812
           # mne-python
           nibabel
+          # View niftis!
+          ipyniivue
           neo
           
           # Neurosimulators
           neuronpy
-          nest
+          # nest with MPI is incompatible with Jupyter:
+          # https://www.nest-simulator.org/pynest-api/_modules/nest.html
+          # see pynestkernel comments about their workaround
+          (nest.override {
+            withMpi = false;
+          })
           brian2
         ]));
       in {