manatoboys commited on
Commit
2d2be79
·
1 Parent(s): f8128c7
Files changed (2) hide show
  1. app.py +1 -1
  2. main.py +3 -3
app.py CHANGED
@@ -33,7 +33,7 @@ def company_scoring(company_name):
33
  header = header5
34
 
35
  header = ', '.join(header)
36
- if related_url_list is list:
37
  related_url_list = ','.join(related_url_list)
38
  if group == 'Group 1-2' or group == 'Group 5':
39
  result = company_name + ',' + group + ',' + related_url_list
 
33
  header = header5
34
 
35
  header = ', '.join(header)
36
+ if isinstance(related_url_list, list):
37
  related_url_list = ','.join(related_url_list)
38
  if group == 'Group 1-2' or group == 'Group 5':
39
  result = company_name + ',' + group + ',' + related_url_list
main.py CHANGED
@@ -184,10 +184,10 @@ def main(company_name, docID):
184
  url = [PDF]
185
  group, related_url_list, unrelated_url_list, other_url_list = group2_classification(company_name)
186
  logger.info(f"related_url_list: {related_url_list}")
187
- if related_url_list is not list:
188
- related_url_list = [PDF]
189
- else:
190
  related_url_list += url
 
 
191
  logger.info(f'{company_name}: {group}')
192
  if group == 'Group 2':
193
  scoring_result, urls = group2_scoring.scoring(company_name)
 
184
  url = [PDF]
185
  group, related_url_list, unrelated_url_list, other_url_list = group2_classification(company_name)
186
  logger.info(f"related_url_list: {related_url_list}")
187
+ if isinstance(related_url_list, list):
 
 
188
  related_url_list += url
189
+ else:
190
+ related_url_list = [PDF]
191
  logger.info(f'{company_name}: {group}')
192
  if group == 'Group 2':
193
  scoring_result, urls = group2_scoring.scoring(company_name)