Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Jan 25, 2025
1 parent b05a9c3 commit 896f97e
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions examples/llm/tech_qa.py
Original file line number Diff line number Diff line change
Expand Up @@ -89,8 +89,8 @@ def make_dataset(args):
rawset = load_dataset('rojagtap/tech-qa', trust_remote_code=True)
data_lists = {"train": [], "validation": [], "test": []}
kg_maker = TXT2KG(NVIDIA_NIM_MODEL=args.NV_NIM_MODEL,
NVIDIA_API_KEY=args.NV_NIM_KEY,
chunk_size=args.chunk_size)
NVIDIA_API_KEY=args.NV_NIM_KEY,
chunk_size=args.chunk_size)
triples = []
for split_str in data_lists.keys():
if split_str == "test":
Expand Down

0 comments on commit 896f97e

Please sign in to comment.