Fix performance logic for text to multioption

huridocs · Nov 11, 2024 · bd53870 · bd53870
1 parent 8d85621
commit bd53870
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/...ble_entity_extractor/extractors/text_to_multi_option_extractor/TextToMultiOptionMethod.py b/...ble_entity_extractor/extractors/text_to_multi_option_extractor/TextToMultiOptionMethod.py
@@ -57,9 +57,7 @@ def performance(self, extraction_data: ExtractionData) -> float:
 
         self.train(performance_train_set)
 
-        prediction_samples = [
-            PredictionSample(segment_selector_texts=x.segment_selector_texts) for x in performance_test_set.samples
-        ]
+        prediction_samples = [PredictionSample(source_text=x.labeled_data.source_text) for x in performance_test_set.samples]
         predictions = self.predict(prediction_samples)
 
         self.remove_model()