From a57fd52a5b7b927e97ffc2dfda3441882664c0c2 Mon Sep 17 00:00:00 2001
From: Shaojun Liu <61072813+liu-shaojun@users.noreply.github.com>
Date: Mon, 18 Mar 2024 13:56:34 +0800
Subject: [PATCH] pip install notebook (#10444)

---
 .github/workflows/llm_unit_tests.yml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/.github/workflows/llm_unit_tests.yml b/.github/workflows/llm_unit_tests.yml
index 783606c7..c76843ef 100644
--- a/.github/workflows/llm_unit_tests.yml
+++ b/.github/workflows/llm_unit_tests.yml
@@ -268,6 +268,7 @@ jobs:
           python -m pip install --upgrade pip
           python -m pip install --upgrade setuptools
           python -m pip install --upgrade wheel
+          python -m pip install --upgrade notebook
 
       - name: Download llm binary
         uses: ./.github/actions/llm/download-llm-binary