From f07cd957477d8f9251bd8780dcd77e2200151026 Mon Sep 17 00:00:00 2001
From: Mattia <32590908+Uzarel@users.noreply.github.com>
Date: Thu, 25 Jan 2024 18:12:29 +0100
Subject: [PATCH] Update testset_generation.md

Explicitly added distributions to dataset generation, otherwise it won't work
---
 docs/getstarted/testset_generation.md | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/docs/getstarted/testset_generation.md b/docs/getstarted/testset_generation.md
index d4652f02d..8735b3d12 100644
--- a/docs/getstarted/testset_generation.md
+++ b/docs/getstarted/testset_generation.md
@@ -46,8 +46,7 @@ from ragas.testset.evolutions import simple, reasoning, multi_context
 generator = TestsetGenerator.with_openai()
 
 # generate testset
-testset = generator.generate_with_langchain_docs(documents, test_size=10)
-testset.to_pandas()
+testset = generator.generate_with_langchain_docs(documents, test_size=10, distributions={simple: 0.5, reasoning: 0.25, multi_context: 0.25})
 ```
 
 Subsequently, we can export the results into a Pandas DataFrame.