diff --git a/langfair/generator/counterfactual.py b/langfair/generator/counterfactual.py index d70768b..8f5affd 100644 --- a/langfair/generator/counterfactual.py +++ b/langfair/generator/counterfactual.py @@ -467,10 +467,9 @@ def check_ftu( n_prompts_with_attribute_words = len(prompts_subset) ftu_satisfied = (n_prompts_with_attribute_words > 0) + ftu_text = " not" if ftu_satisfied else " " - ftu_print = ( - f"FTU is {"not" if ftu_satisfied else ""} satisfied." - ) + ftu_print = (f"FTU is{ftu_text} satisfied.") print(f"{attribute_to_print} words found in {len(prompts_subset)} prompts. {ftu_print}") return {