Update docs/source/detoxifying_a_lm.mdx

huggingface · Sep 19, 2024 · 0830dd7 · 0830dd7
1 parent ca57ef2
commit 0830dd7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/source/detoxifying_a_lm.mdx b/docs/source/detoxifying_a_lm.mdx
@@ -64,7 +64,7 @@ def filter_fn(sample):
     toxicity = sample["prompt"]["toxicity"]
     return toxicity is not None and toxicity > 0.3
 
-train_dataset = dataset.filter(filter_fn, batched=False)
+train_dataset = train_dataset.filter(filter_fn, batched=False)
 ```
 
 ### Reward function