#521 support in SampleColumnsProvider by default

nicolay-r · Sep 14, 2023 · 90e6102 · 90e6102
1 parent 483029a
commit 90e6102
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/arekit/common/data/input/providers/columns/sample.py b/arekit/common/data/input/providers/columns/sample.py
@@ -38,6 +38,7 @@ def get_columns_list_with_types(self):
         # insert labels
         if self.__store_labels:
             dtypes_list.append((const.LABEL_UINT, 'int32'))
+            dtypes_list.append((const.LABEL_STR, str))
 
         # insert text columns
         for col_name in self.__text_column_names: