add missing argument and update language

cvs-health · Jan 1, 2025 · f14c543 · f14c543
1 parent ffb3109
commit f14c543
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/evaluations/text_generation/counterfactual_metrics_demo.ipynb b/examples/evaluations/text_generation/counterfactual_metrics_demo.ipynb
@@ -439,7 +439,7 @@
     "    attribute='race',\n",
     "    subset_prompts=True\n",
     ")\n",
-    "race_prompts = pd.DataFrame(ftu_result[\"data\"]).rename({'attribute_words': 'race_words'})\n",
+    "race_prompts = pd.DataFrame(ftu_result[\"data\"]).rename(columns={'attribute_words': 'race_words'})\n",
     "race_prompts.tail(5)"
    ]
   },
@@ -481,7 +481,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "Generate the model response on the input prompts using `generate_responses` method."
+    "Create counterfactual input prompts and generate corresponding LLM responses  using `generate_responses` method."
    ]
   },
   {