The world's most clever kitty cat
Stable sorting for /weights, less noisy output
+8
-4
+8
-4
src/cmd/weights.rs
+8
-4
src/cmd/weights.rs
···
20
20
21
21
brain.get_weights(token).map(|edges| {
22
22
let sep = String::from("\n");
23
-
let all_weights = edges
24
-
.iter_weights()
23
+
let mut all_weights = edges.iter_weights().collect::<Vec<_>>();
24
+
25
+
all_weights.sort_by_key(|(_, w, _)| *w);
26
+
27
+
let formatted_weights = all_weights
28
+
.into_iter()
25
29
.map(|(token, weight, chance)| {
26
30
let token_fmt = format_token(token);
27
-
format!("- **{token_fmt}**: {:.1}% ({weight})", chance * 100.0)
31
+
format!("{token_fmt}: {:.1}% ({weight})", chance * 100.0)
28
32
})
29
33
.intersperse(sep)
30
34
.collect::<String>();
31
35
32
-
format!("Weights for {token}:\n{all_weights}")
36
+
format!("Weights for {token}:\n{formatted_weights}")
33
37
})
34
38
}
35
39