refactor

src/colpali/visualizer.py

@@ -4,15 +4,15 @@ ColPali Visualization Module
 Generates attention/saliency maps to visualize which parts of the document
 are most relevant to a query.
 """
+import logging
+from typing import List, Dict, Any, Optional
 
 import torch
 import numpy as np
-from PIL import Image, ImageDraw, ImageFont
-from typing import List, Dict, Any, Optional
 import matplotlib.pyplot as plt
 import matplotlib.patches as patches
+from PIL import Image, ImageDraw, ImageFont
 from matplotlib.colors import LinearSegmentedColormap
-import logging
 
 
 logger = logging.getLogger(__name__)
@@ -231,6 +231,3 @@ def visualize_retrieval_results(
         plt.show()
     
     plt.close()
-
-
-