Fix hf example

hegelai · Jul 24, 2023 · 0bd6ecf · 0bd6ecf
1 parent 71ae416
commit 0bd6ecf
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/examples/.DS_Store b/examples/.DS_Store
diff --git a/examples/prompttests/test_huggingface_hub.py b/examples/prompttests/test_huggingface_hub.py
@@ -8,7 +8,7 @@
 import jinja2
 import prompttools.prompttest as prompttest
 from prompttools.utils import similarity
-
+from prompttools.prompttest.threshold_type import ThresholdType
 from prompttools.mock.mock import mock_hf_completion_fn
 from huggingface_hub.inference_api import InferenceApi
 
@@ -37,6 +37,8 @@ def create_prompt():
     eval_fn=similarity.evaluate,
     prompts=[create_prompt()],
     expected=["George Washington"],
+    threshold=1.0,
+    threshold_type=ThresholdType.MAXIMUM
 )
 def completion_fn(prompt: str):
     response = None