Clean up imports and remove unused imports in evaluation steps

Refactor import statements in eval_retrieval.py and eval_visualisation.py to:
- Remove unused imports
- Organize imports consistently
- Simplify import statements

Author: strickvl

llm-complete-guide/steps/eval_retrieval.py

@@ -14,16 +14,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import json
 import logging
-from typing import Annotated, List, Tuple, Dict, Callable, Any, Optional
-from multiprocessing import Pool, cpu_count
 from functools import partial
-from tenacity import retry, stop_after_attempt, wait_exponential
-from tqdm import tqdm
-from concurrent.futures import ThreadPoolExecutor
-import json
+from multiprocessing import Pool, cpu_count
+from typing import Annotated, Any, Callable, Dict, List, Optional, Tuple
 
 from datasets import load_dataset
+from tenacity import retry, stop_after_attempt, wait_exponential
 from utils.llm_utils import (
     find_vectorstore_name,
     get_db_conn,
@@ -32,7 +30,7 @@
     get_topn_similar_docs,
     rerank_documents,
 )
-from zenml import step, get_step_context
+from zenml import step
 from zenml.logger import get_logger
 
 logger = get_logger(__name__)

llm-complete-guide/steps/eval_visualisation.py

@@ -18,7 +18,7 @@
 import matplotlib.pyplot as plt
 import numpy as np
 from PIL import Image
-from zenml import ArtifactConfig, get_step_context, step, log_metadata
+from zenml import ArtifactConfig, get_step_context, log_metadata, step
 
 
 def create_image(