From 4d17fa9953c3cbdda8cdab5cfa8c39624cf81578 Mon Sep 17 00:00:00 2001
From: HuXiangkun <xkhu17@fudan.edu.cn>
Date: Thu, 5 Sep 2024 03:05:19 +0000
Subject: [PATCH] bug fix

---
 pyproject.toml                        | 2 +-
 refchecker/extractor/llm_extractor.py | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index 786751c..728b879 100644
--- 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 <xiangkhu@amazon.com>",
diff --git a/refchecker/extractor/llm_extractor.py b/refchecker/extractor/llm_extractor.py
index aad520a..79a26b7 100644
--- 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.