#376 related, #459 to support frames; #502 related

nicolay-r · Jul 22, 2023 · 539c8f5 · 539c8f5
1 parent 194e601
commit 539c8f5
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/arekit/common/data/input/providers/columns/sample.py b/arekit/common/data/input/providers/columns/sample.py
@@ -47,6 +47,10 @@ def get_columns_list_with_types(self):
         dtypes_list.append((const.S_IND, 'int32'))
         dtypes_list.append((const.T_IND, 'int32'))
 
+        # opinion-extraction task related fields
+        dtypes_list.append((const.OPINION_ID, 'int32'))
+        dtypes_list.append((const.OPINION_LINKAGE_ID, 'int32'))
+
         return dtypes_list
 
     def set_text_column_names(self, text_column_names):