Skip to content

Commit

Permalink
Fix bug with test_correct_api_dataset_with_text_preprocessing by upda…
Browse files Browse the repository at this point in the history
…te col_type regex rule for str*
  • Loading branch information
aPovidlo committed Aug 16, 2024
1 parent 16285df commit 36f994c
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion fedot/preprocessing/preprocessing.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
import re
from copy import copy
from typing import Optional, Union

Expand Down Expand Up @@ -568,7 +569,7 @@ def reduce_mem_usage_np(arr, initial_types):
col = col.astype(init_type)
col_type = col.dtype.name

if col_type not in ['object', 'str32', 'str96', 'str128', 'str160', 'str384']:
if col_type not in ['object'] and not bool(re.match(r'str\d*$', col_type)):
c_min = col.max()
c_max = col.max()

Expand Down

0 comments on commit 36f994c

Please sign in to comment.