From d2a2af3ee242bdcbb41d24ca036085e8488263c0 Mon Sep 17 00:00:00 2001
From: Aaron <29749331+aarnphm@users.noreply.github.com>
Date: Wed, 4 Oct 2023 19:21:14 -0400
Subject: [PATCH] fix: import nbformat for playground

Signed-off-by: Aaron <29749331+aarnphm@users.noreply.github.com>
---
 openllm-python/src/openllm/cli/extension/playground.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/openllm-python/src/openllm/cli/extension/playground.py b/openllm-python/src/openllm/cli/extension/playground.py
index 1fd2c0fa..9183facb 100644
--- a/openllm-python/src/openllm/cli/extension/playground.py
+++ b/openllm-python/src/openllm/cli/extension/playground.py
@@ -9,6 +9,7 @@ import tempfile
 import typing as t
 
 import click
+import nbformat
 import yaml
 
 from openllm import playground
@@ -19,7 +20,6 @@ from openllm_core.utils import is_notebook_available
 
 if t.TYPE_CHECKING:
   import jupytext
-  import nbformat
 
   from openllm_core._typing_compat import DictStrAny