Skip to content

Commit 28068a8

Browse files
committed
Fix unit test
1 parent 28439c4 commit 28068a8

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/tests/trainable_entity_extractor/use_cases/test_extractor_text_to_multi_option.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -24,10 +24,10 @@ def test_get_text_multi_option_suggestions(self):
2424
options = [Option(id="1", label="abc"), Option(id="2", label="dfg"), Option(id="3", label="hij")]
2525

2626
values_1 = [Option(id="1", label="abc"), Option(id="2", label="dfg")]
27-
labeled_data_1 = LabeledData(language_iso="en", values=values_1, source_text="abc dfg")
27+
labeled_data_1 = LabeledData(language_iso="en", values=values_1, source_text="foo abc dfg")
2828

2929
values_2 = [Option(id="2", label="dfg"), Option(id="3", label="hij")]
30-
labeled_data_2 = LabeledData(language_iso="en", values=values_2, source_text="dfg hij")
30+
labeled_data_2 = LabeledData(language_iso="en", values=values_2, source_text="foo dfg hij")
3131

3232
sample = [TrainingSample(labeled_data=labeled_data_1), TrainingSample(labeled_data=labeled_data_2)]
3333
extraction_data = ExtractionData(

0 commit comments

Comments
 (0)