fix: import errors and make experimental runnable (#1078)

Author: jjmachan

src/experimental/ragas_experimental/testset/generators/base.py

@@ -48,8 +48,7 @@ def generate(
         docs: t.Sequence[Document],
         test_size: int,
         distribution: QADistribution,
-    ) -> TestDataset:
-        ...
+    ) -> TestDataset: ...
 
     def generate_with_langchain_docs(
         self,

src/experimental/ragas_experimental/testset/generators/simple.py

@@ -13,7 +13,6 @@
 )
 from ragas_experimental.testset.generators import (
     QADistribution,
-    TestDataset,
     TestGenerator,
 )
 from ragas_experimental.testset.graph import Node, NodeLevel
@@ -35,6 +34,7 @@
 from ragas.executor import Executor
 from ragas.llms.base import llm_factory
 from ragas.utils import check_if_sum_is_close
+from ragas.testset.generator import TestDataset
 
 abstract_qa = AbstractQA(distribution=DEFAULT_DISTRIBUTION)
 comparative_qa = ComparativeAbstractQA(distribution=DEFAULT_DISTRIBUTION)

src/experimental/ragas_experimental/testset/questions/__init__.py

@@ -4,6 +4,7 @@
 )
 from ragas_experimental.testset.questions.base import (
     DEFAULT_DISTRIBUTION,
+    QAC,
     QAGenerator,
     QuestionLength,
     QuestionStyle,
@@ -18,6 +19,7 @@
     "QAGenerator",
     "QuestionStyle",
     "QuestionLength",
+    "QAC",
     "DEFAULT_DISTRIBUTION",
     "StyleLengthDistribution",
 ]