From c535a4ab3973842f63e552d39d4fe7e0631cf732 Mon Sep 17 00:00:00 2001
From: jjmachan <jamesjithin97@gmail.com>
Date: Thu, 2 Nov 2023 15:18:14 +0530
Subject: [PATCH] docs: add import statement in testset generation

---
 docs/getstarted/testset_generation.md | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/docs/getstarted/testset_generation.md b/docs/getstarted/testset_generation.md
index 1d5d02413..d78ede3ed 100644
--- a/docs/getstarted/testset_generation.md
+++ b/docs/getstarted/testset_generation.md
@@ -33,6 +33,8 @@ We will now import and use Ragas' `Testsetgenerator` to promptly generate a synt
 
 ```{code-block} python
 :caption: Create 10 samples using default configuration
+from ragas.testset import TestsetGenerator
+
 testsetgenerator = TestsetGenerator.from_default()
 test_size = 10
 testset = testsetgenerator.generate(documents, test_size=test_size)