Skip to content

Commit e7e4aaa

Browse files
Update testset.py
1 parent 085abe6 commit e7e4aaa

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

‎giskard/rag/testset.py‎

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,6 @@
1515

1616
from ..core.suite import Suite
1717
from ..datasets.base import Dataset
18-
from ..llm.client import get_default_client
1918
from ..testing.tests.llm import test_llm_correctness
2019

2120
logger = logging.getLogger(__name__)
@@ -156,7 +155,12 @@ def push_to_hf_hub(
156155
template = template_path.read_text()
157156

158157
# Make and push the dataset card
159-
config = {"metadata": get_default_client().get_config()}
158+
try:
159+
from ..llm.client import get_default_client
160+
161+
config = {"metadata": get_default_client().get_config()}
162+
except Exception:
163+
config = None
160164
content = template.format(repo_id=repo_id, num_items=len(self._dataframe), config=json.dumps(config, indent=4))
161165
return DatasetCard(content=content).push_to_hub(repo_id=repo_id, token=token, repo_type="dataset")
162166

0 commit comments

Comments
 (0)