From cf5a6d1ab7b6cc5bce5e58744e31ffe409eb2776 Mon Sep 17 00:00:00 2001
From: Hubert <42952108+yingfhu@users.noreply.github.com>
Date: Wed, 8 Nov 2023 17:58:49 +0800
Subject: [PATCH] [Fix] fix unnecessary import and update requirements (#555)

---
 opencompass/datasets/cibench.py | 3 +--
 requirements/extra.txt          | 3 +++
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/opencompass/datasets/cibench.py b/opencompass/datasets/cibench.py
index 86f08e75..9c57f88b 100644
--- a/opencompass/datasets/cibench.py
+++ b/opencompass/datasets/cibench.py
@@ -7,7 +7,6 @@ from typing import List, Optional
 import numpy as np
 from datasets import Dataset
 
-from opencompass.lagent.actions.ipython_interpreter import extract_code
 from opencompass.openicl.icl_evaluator import BaseEvaluator
 from opencompass.registry import LOAD_DATASET
 
@@ -172,7 +171,7 @@ class CIBenchEvaluator(BaseEvaluator):
                 'the conversion processes.')
 
         check_jupytext()
-
+        from opencompass.lagent.actions.ipython_interpreter import extract_code
         for idx, (example_origin_prompt,
                   example_steps) in enumerate(zip(origin_prompt, steps)):
             markdown_lines = []
diff --git a/requirements/extra.txt b/requirements/extra.txt
index f5f709ce..d26a4f6e 100644
--- a/requirements/extra.txt
+++ b/requirements/extra.txt
@@ -1 +1,4 @@
 faiss_gpu==1.7.2
+jupyter
+lagent
+scikit-image