refactor

Lightning-AI · Dec 20, 2024 · a33d69a · a33d69a
1 parent c6c433e
commit a33d69a
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/src/torchmetrics/functional/multimodal/clip_score.py b/src/torchmetrics/functional/multimodal/clip_score.py
@@ -157,14 +157,8 @@ def _clip_score_update(
     source_modality = _detect_modality(source)
     target_modality = _detect_modality(target)
 
-    if source_modality == "image":
-        source_data = _process_image_data(source)
-    else:
-        source_data = _process_text_data(source)
-    if target_modality == "image":
-        target_data = _process_image_data(target)
-    else:
-        target_data = _process_text_data(target)
+    source_data = _process_image_data(source) if source_modality == "image" else _process_text_data(source) 
+    target_data = _process_image_data(target) if target_modality == "image" else _process_text_data(target)
 
     # Verify matching lengths
     if len(source_data) != len(target_data):