From 06ccfb63e45ce72227b9337e0ceb699207d999cc Mon Sep 17 00:00:00 2001
From: Vincent Roseberry <rosbo@google.com>
Date: Wed, 7 Apr 2021 17:39:45 +0000
Subject: [PATCH] Remove pins for jupyter packages.

- Use simply the version from our DLVM base image.
- Removed the meta-package `jupyter`. It references packages already
installed.

This ensures we have the jupyter packages in sync with DLVM and ensures
compatibility with the Cloud AI Notebook environment.
---
 Dockerfile | 15 +++++----------
 1 file changed, 5 insertions(+), 10 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index f5e4d0c2..9706d5fd 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -304,17 +304,14 @@ RUN pip install bleach && \
     pip install decorator && \
     pip install entrypoints && \
     pip install html5lib && \
-    # Latest version breaks nbconvert: https://github.com/ipython/ipykernel/issues/422
-    pip install ipykernel==5.1.1 && \
+    pip install ipykernel && \
     pip install ipython && \
     pip install ipython-genutils && \
     pip install ipywidgets && \
     pip install isoweek && \
-    # Lastest version breaks ipython. https://github.com/ipython/ipython/issues/12740
-    pip install jedi==0.17.2 && \
+    pip install jedi && \
     pip install Jinja2 && \
     pip install jsonschema && \
-    pip install jupyter && \
     pip install jupyter-client && \
     pip install jupyter-console && \
     pip install jupyter-core && \
@@ -322,7 +319,8 @@ RUN pip install bleach && \
     pip install mistune && \
     pip install nbconvert && \
     pip install nbformat && \
-    pip install notebook==5.5.0 && \
+    pip install notebook && \
+    pip install papermill && \
     pip install olefile && \
     pip install kornia && \
     pip install pandas_summary && \
@@ -342,9 +340,7 @@ RUN pip install bleach && \
     pip install qtconsole && \
     pip install six && \
     pip install terminado && \
-    # Latest version (6.0) of tornado breaks Jupyter notebook:
-    # https://github.com/jupyter/notebook/issues/4439
-    pip install tornado==5.0.2 && \
+    pip install tornado && \
     pip install tqdm && \
     pip install traitlets && \
     pip install wcwidth && \
@@ -411,7 +407,6 @@ RUN pip install flashtext && \
     pip install nnabla && \
     pip install vowpalwabbit && \
     # papermill can replace nbconvert for executing notebooks
-    pip install papermill && \
     pip install cloud-tpu-client && \
     pip install tensorflow-cloud && \
     pip install tensorflow-datasets && \