update: tests

wandb · Oct 30, 2024 · 342ba02 · 342ba02
1 parent c27fb4c
commit 342ba02
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/hemm/tests/test_prompt_alignment_eval.py b/hemm/tests/test_prompt_alignment_eval.py
@@ -13,9 +13,9 @@ def test_prompt_alignment_evaluation():
         enable_cpu_offfload=False,
     )
     clip_scorer = CLIPScoreMetric(
-        clip_model_name_or_path="openai/clip-vit-base-patch16"
+        model_name="openai/clip-vit-base-patch16"
     )
-    clip_iqa_scorer = CLIPImageQualityScoreMetric(clip_model_name_or_path="clip_iqa")
+    clip_iqa_scorer = CLIPImageQualityScoreMetric(model_name="clip_iqa")
     dataset = weave.ref("parti-prompts:v0").get().rows[:2]
     evaluation = weave.Evaluation(
         dataset=dataset, scorers=[clip_scorer, clip_iqa_scorer]