Notebook updates

ci/test/notebooks.sh

@@ -6,9 +6,23 @@ set -o pipefail
 export LIBCUDF_KERNEL_CACHE_PATH=${WORKSPACE}/.jitcache
 
 source /opt/conda/bin/activate rapids
+
+# PyTorch is intentionally excluded from our Docker images due
+# to its size, but some notebooks still depend on it.
+case "${CUDA_VER}" in
+"10.1" | "10.2")
+    conda install -y -c pytorch pytorch
+    ;;
+*)
+    echo "Unsupported CUDA version for pytorch."
+    echo "Not installing pytorch."
+    ;;
+esac
+
+
 env
 /test.sh 2>&1 | tee nbtest.log
 EXITCODE=$?
 python /rapids/utils/nbtestlog2junitxml.py nbtest.log
 
-exit ${EXITCODE}
+exit ${EXITCODE}

conda/recipes/rapids-notebook-env/meta.yaml

@@ -42,6 +42,7 @@ requirements:
     - cython {{ cython_version }}
     - dask-labextension
     - dask-ml
+    - filterpy
     - holoviews
     - ipython {{ ipython_version }}
     - jupyter-server-proxy