fix: lint

wandb · Oct 30, 2024 · 4f29ad0 · 4f29ad0
1 parent a2a4331
commit 4f29ad0
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/hemm/tests/test_prompt_alignment_eval.py b/hemm/tests/test_prompt_alignment_eval.py
@@ -12,9 +12,7 @@ def test_prompt_alignment_evaluation():
         diffusion_model_name_or_path="CompVis/stable-diffusion-v1-4",
         enable_cpu_offfload=False,
     )
-    clip_scorer = CLIPScoreMetric(
-        model_name="openai/clip-vit-base-patch16"
-    )
+    clip_scorer = CLIPScoreMetric(model_name="openai/clip-vit-base-patch16")
     clip_iqa_scorer = CLIPImageQualityScoreMetric(model_name="clip_iqa")
     dataset = weave.ref("parti-prompts:v0").get().rows[:2]
     evaluation = weave.Evaluation(