nam194 commited on
Commit
aa67117
1 Parent(s): c218615

Update parse_info.py

Browse files
Files changed (1) hide show
  1. parse_info.py +3 -0
parse_info.py CHANGED
@@ -55,6 +55,9 @@ skill_list = [i.replace("\n","") for i in skill_list["Skill"].to_list()]
55
  def parse_skill(inp: List) -> list:
56
  res = []
57
  for i, _ in enumerate(inp):
 
 
 
58
  inp[i] = parse_string(inp[i])
59
  for ngram in Counter(map(' '.join, everygrams(" ".join(inp).split(), 1, 3))).keys():
60
  if ngram in skill_list:
 
55
  def parse_skill(inp: List) -> list:
56
  res = []
57
  for i, _ in enumerate(inp):
58
+ if "," in _:
59
+ _ = [j.strip() for j in _.split(",")]
60
+ inp.extend(_)
61
  inp[i] = parse_string(inp[i])
62
  for ngram in Counter(map(' '.join, everygrams(" ".join(inp).split(), 1, 3))).keys():
63
  if ngram in skill_list: