style: apply consistent formatting to import statements across multiple files

routers/visualize.py

@@ -5,10 +5,16 @@ import os
 from fastapi import APIRouter, HTTPException
 from fastapi.responses import FileResponse
 
-from schemas.visualize import (VisualizeHeatmapRequest,
-                               VisualizeMeanDiffRequest, VisualizePCARequest)
-from utils.visualize_pca import (run_visualize_heatmap,
-                                 run_visualize_mean_diff, run_visualize_pca)
+from schemas.visualize import (
+    VisualizeHeatmapRequest,
+    VisualizeMeanDiffRequest,
+    VisualizePCARequest,
+)
+from utils.visualize_pca import (
+    run_visualize_heatmap,
+    run_visualize_mean_diff,
+    run_visualize_pca,
+)
 
 logger = logging.getLogger(__name__)
 logger.setLevel(logging.INFO)

utils/visualize_pca.py

@@ -7,8 +7,7 @@ from typing import List, Optional, Tuple, Union
 
 import matplotlib
 import torch
-from optipfair.bias import (visualize_heatmap, visualize_mean_differences,
-                            visualize_pca)
+from optipfair.bias import visualize_heatmap, visualize_mean_differences, visualize_pca
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
 matplotlib.use("Agg")  # Use 'Agg' backend for non-GUI environments