Spaces:
Running
Running
fix argsort
Browse files- 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:
|
36 |
results_as_text = dict()
|
37 |
-
for i in
|
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]
|