Gainward777 commited on
Commit
65fbefb
·
verified ·
1 Parent(s): 4ae1ffc

Update preprocess/preprocess.py

Browse files
Files changed (1) hide show
  1. preprocess/preprocess.py +1 -1
preprocess/preprocess.py CHANGED
@@ -41,7 +41,7 @@ class Preprocessor():
41
  result['brand'].append(i['brand'])
42
  else: result['brand'].append(None)
43
  result['name'].append(i['name'])
44
- drink_type=get_type(i, self.types)
45
  if drink_type is None:
46
  drink_type=check_spark(i)
47
  if drink_type is None:
 
41
  result['brand'].append(i['brand'])
42
  else: result['brand'].append(None)
43
  result['name'].append(i['name'])
44
+ drink_type=get_type(i, self.long_types_list)
45
  if drink_type is None:
46
  drink_type=check_spark(i)
47
  if drink_type is None: