[fix] Add matplotlib as dependency and initial import

.tools/actions_utils/environment.gpu.yml

@@ -21,4 +21,5 @@ dependencies:
   - nvidia-cudnn-cu11==8.6.0.163 # if GPU_FLAG==1
   - nbformat
   - jupyterlab
-  - ipywidgets
+  - ipywidgets
+  - matplotlib

.tools/notebook_autoconversion/code_utils_one_cell.py

@@ -16,6 +16,7 @@
 
 ipywidget_imported_code = ("import ipywidgets as widgets\n" 
                         "from IPython.display import Markdown, display, clear_output\n"
+                        "from matplotlib import pyplot as plt\n"
                         "import yaml as yaml_library\n"
                         "import os\n"
                         "\n"

Dockerfile

@@ -107,16 +107,19 @@ RUN pip install --upgrade pip && \
     fi && \
     if [  "$(printf '%s\n' "3.7" "${PYTHON_VERSION}" | sort -V | head -n1)" = "3.7" ] ; then \
         # For Python between 3.7 and 3.11
-        pip install  ipywidgets==8.1.0 && \ 
-        pip install  jupyterlab==3.6.0 ; \ 
+        pip install ipywidgets==8.1.0 && \ 
+        pip install jupyterlab==3.6.0 && \ 
+        pip install matplotlib==3.6.3 ; \  
     elif [  "$(printf '%s\n' "3.5" "${PYTHON_VERSION}" | sort -V | head -n1)" = "3.5" ] ; then \
         # For Python 3.4 or lower
-        pip install  ipywidgets==7.8.1 && \ 
-        pip install  jupyterlab==0.23.0 ; \ 
+        pip install ipywidgets==7.8.1 && \ 
+        pip install jupyterlab==0.23.0 && \ 
+        pip install matplotlib==2.1.0 ; \ 
     else \
         # For Python 3.5 and 3.6
-        pip install  ipywidgets==8.0.0a0 && \ 
-        pip install  jupyterlab==2.3.2 ; \ 
+        pip install ipywidgets==8.0.0a0 && \ 
+        pip install jupyterlab==2.3.2 && \ 
+        pip install matplotlib==3.0.3 ; \ 
     fi
 
 # Custom cache invalidation

Dockerfile.gpu

@@ -134,15 +134,18 @@ RUN pip install --upgrade pip && \
     if [  "$(printf '%s\n' "3.7" "${PYTHON_VERSION}" | sort -V | head -n1)" = "3.7" ] ; then \
         # For Python between 3.7 and 3.11
         pip install ipywidgets==8.1.0 && \ 
-        pip install jupyterlab==3.6.0 ; \ 
+        pip install jupyterlab==3.6.0 && \ 
+        pip install matplotlib==3.6.3 ; \  
     elif [  "$(printf '%s\n' "3.5" "${PYTHON_VERSION}" | sort -V | head -n1)" = "3.5" ] ; then \
         # For Python 3.4 or lower
         pip install ipywidgets==7.8.1 && \ 
-        pip install jupyterlab==0.23.0 ; \ 
+        pip install jupyterlab==0.23.0 && \ 
+        pip install matplotlib==2.1.0 ; \ 
     else \
         # For Python 3.5 and 3.6
         pip install ipywidgets==8.0.0a0 && \ 
-        pip install jupyterlab==2.3.2 ; \ 
+        pip install jupyterlab==2.3.2 && \ 
+        pip install matplotlib==3.0.3 ; \ 
     fi
 
 # Custom cache invalidation