more explicit template filter

sgbaird · Jan 28, 2025 · 942b172 · 942b172
1 parent c5e1131
commit 942b172
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/honegumi/ax/main.py.jinja b/src/honegumi/ax/main.py.jinja
@@ -332,9 +332,9 @@ for i in range({{ num_iter }}):
     {{ indent }}    x1, x2{% if composition_constraint %}, x3{% endif %}{% if categorical %}, c1{% endif %}{% if task == "Multi" %}, task{% endif %}
     {{ indent }}    )
     {{ indent }}ax_client.complete_trial(trial_index=trial_index, raw_data=results)
-{% if objective == "Multi" and task != "Multi" -%}
+{% if objective == "Multi" and task == "Single" -%}
 pareto_results = ax_client.get_pareto_optimal_parameters()
-{% else %}
+{% elif objective == "Single" %}
 best_parameters, metrics = ax_client.get_best_parameters()
 {%- endif %}