Skip to content

Commit 3a62f75

Browse files
author
Dash
committed
style(testset_generator): rename propose to proposal
1 parent b202099 commit 3a62f75

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

src/ragas/testset/testset_generator.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@
7878
],
7979
)
8080

81-
Propose = namedtuple("Propose", ["question", "text_chunk"])
81+
Proposal = namedtuple("Proposal", ["question", "text_chunk"])
8282

8383

8484
@dataclass
@@ -302,9 +302,9 @@ def _embed_nodes(self, nodes: t.List[BaseNode]) -> t.Dict[str, t.List[float]]:
302302

303303
return embeddings
304304

305-
def _make_propose(
305+
def _make_proposal(
306306
self, cur_node: BaseNode, neighbor_nodes: t.List[BaseNode], evolve_type: str
307-
) -> t.Union[Propose, None]:
307+
) -> t.Union[Proposal, None]:
308308
# Append multiple nodes randomly to remove chunking bias
309309
size = self.rng.integers(1, 3)
310310
nodes = (
@@ -364,7 +364,7 @@ def _make_propose(
364364
else:
365365
question = self._compress_question(question=question)
366366

367-
return Propose(question=question, text_chunk=text_chunk)
367+
return Proposal(question=question, text_chunk=text_chunk)
368368

369369
def generate(
370370
self,
@@ -414,20 +414,20 @@ def generate(
414414

415415
neighbor_nodes = doc_nodes_map[curr_node.source_node.node_id]
416416

417-
propose = None
417+
proposal = None
418418
try:
419-
propose = self._make_propose(
419+
proposal = self._make_proposal(
420420
curr_node, neighbor_nodes, evolve_type
421421
)
422422
except Exception as e:
423423
err_cause = e.__cause__
424424
if not isinstance(err_cause, retry_errors):
425425
raise e
426426

427-
if propose is None:
427+
if proposal is None:
428428
continue
429-
question = propose.question
430-
text_chunk = propose.text_chunk
429+
question = proposal.question
430+
text_chunk = proposal.text_chunk
431431

432432
is_valid_question = self._filter_question(question)
433433
if is_valid_question:

0 commit comments

Comments
 (0)