bug fix

amazon-science · Sep 5, 2024 · 4d17fa9 · 4d17fa9
1 parent ba57354
commit 4d17fa9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "refchecker"
-version = "0.2.6"
+version = "0.2.7"
 description = "RefChecker provides automatic checking pipeline for detecting fine-grained hallucinations generated by Large Language Models."
 authors = [
     "Xiangkun Hu <[email protected]>",

diff --git a/refchecker/extractor/llm_extractor.py b/refchecker/extractor/llm_extractor.py
@@ -102,6 +102,7 @@ def extract_claim_triplets(
         batch_questions=None, 
         max_new_tokens=500,
         sagemaker_client=None,
+        sagemaker_params=None,
         **kwargs
     ):
         """Extract KG triplets from the response text.