[WIP] Update

for-ai · Oct 5, 2024 · 9552ce3 · 9552ce3
1 parent 72aa61e
commit 9552ce3
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/analysis/plot_results.py b/analysis/plot_results.py
@@ -44,6 +44,9 @@ def get_args():
     parser_main_results.add_argument("--input_path", type=Path, required=True, help="Path to the results file.")
     parser_main_results.add_argument("--top_ten_only", action="store_true", help="If set, will only show the top-10 of all models.")
     parser_main_results.add_argument("--print_latex", action="store_true", help="If set, print LaTeX table.")
+
+    parser_eng_drop = subparsers.add_parser("eng_drop_line", help="Plot english drop as a line chart.", parents=[shared_args])
+    parser_eng_drop.add_argument("--input_path", type=Path, required=True, help="Path to the results file.")
     # fmt: on
     return parser.parse_args()
 
@@ -53,6 +56,7 @@ def main():
 
     cmd_map = {
         "main_heatmap": plot_main_heatmap,
+        "eng_drop_line": plot_eng_drop_line,
     }
 
     def _filter_args(func, kwargs):