#481 fixed

nicolay-r · May 25, 2023 · f68d25d · f68d25d
1 parent 165c308
commit f68d25d
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/arekit/contrib/prompt/sample.py b/arekit/contrib/prompt/sample.py
@@ -27,14 +27,14 @@ def __init__(self, crop_window_size, label_scaler, text_provider, prompt):
     def _fill_row_core(self, row, text_opinion_linkage, index_in_linked, etalon_label,
                        parsed_news, sentence_ind, s_ind, t_ind):
 
-        row = super(PromptedSampleRowProvider, self)._fill_row_core(row=row,
-                                                                    text_opinion_linkage=text_opinion_linkage,
-                                                                    index_in_linked=index_in_linked,
-                                                                    etalon_label=etalon_label,
-                                                                    parsed_news=parsed_news,
-                                                                    sentence_ind=sentence_ind,
-                                                                    s_ind=s_ind,
-                                                                    t_ind=t_ind)
+        super(PromptedSampleRowProvider, self)._fill_row_core(row=row,
+                                                              text_opinion_linkage=text_opinion_linkage,
+                                                              index_in_linked=index_in_linked,
+                                                              etalon_label=etalon_label,
+                                                              parsed_news=parsed_news,
+                                                              sentence_ind=sentence_ind,
+                                                              s_ind=s_ind,
+                                                              t_ind=t_ind)
         original_text = row[BaseSingleTextProvider.TEXT_A]
         row[BaseSingleTextProvider.TEXT_A] = self.__prompt.format(
             text=original_text,