poltextlab commited on
Commit
61e094a
·
verified ·
1 Parent(s): 555f614

fix argsort

Browse files
Files changed (1) hide show
  1. interfaces/cap_minor.py +2 -2
interfaces/cap_minor.py CHANGED
@@ -32,9 +32,9 @@ domains = {
32
  "local government agenda": "localgovernment"
33
  }
34
 
35
- def convert_minor_to_major(results: str) -> str:
36
  results_as_text = dict()
37
- for i in range(results):
38
  prob = probs[i]
39
  major_code = CAP_MIN_NUM_DICT[i][:-2]
40
  label = CAP_LABEL_NAMES[major_code]
 
32
  "local government agenda": "localgovernment"
33
  }
34
 
35
+ def convert_minor_to_major(results: Sequence[int]) -> Dict[str, float]:
36
  results_as_text = dict()
37
+ for i in results:
38
  prob = probs[i]
39
  major_code = CAP_MIN_NUM_DICT[i][:-2]
40
  label = CAP_LABEL_NAMES[major_code]