Describe the enhancement you'd like

In #347 we pinned ipykernel to 5.5 in order to avoid some regressions in the testing infrastructure. We should upgrade this to 6.0 in order to stay up to date with major versions. This will also entail dropping Python 3.6 support.