gavinzli commited on
Commit
579df96
·
1 Parent(s): 1c5f2e5

Convert single dictionary to list of dictionaries in vectorize function for consistent DataFrame creation

Browse files
Files changed (1) hide show
  1. controllers/vectorizer.py +3 -0
controllers/vectorizer.py CHANGED
@@ -40,6 +40,9 @@ def vectorize(article):
40
  print(article)
41
  print(type(article))
42
  article['id'] = str(article['id'])
 
 
 
43
  df = pd.DataFrame(article)
44
  df = df[['id','site','title','titleCN','category','author','content',
45
  'publishDate','link']]
 
40
  print(article)
41
  print(type(article))
42
  article['id'] = str(article['id'])
43
+ # df = pd.DataFrame(article)
44
+ if isinstance(article, dict):
45
+ article = [article] # Convert single dictionary to list of dictionaries
46
  df = pd.DataFrame(article)
47
  df = df[['id','site','title','titleCN','category','author','content',
48
  'publishDate','link']]