fix:codestype error reportIncompatibleMethodOverride

explodinggradients · Jan 15, 2025 · 3fddbc1 · 3fddbc1
1 parent 3d8751c
commit 3fddbc1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/src/ragas/testset/synthesizers/multi_hop/base.py b/src/ragas/testset/synthesizers/multi_hop/base.py
@@ -158,9 +158,10 @@ def convert_to_scenario(self, data: t.Dict[str, t.Any]) -> MultiHopScenario:
         )
 
     async def _generate_sample(
-        self, scenario: MultiHopScenario, callbacks: Callbacks
+        self, scenario: Scenario, callbacks: Callbacks
     ) -> SingleTurnSample:
-
+        if not isinstance(scenario, MultiHopScenario):
+            logger.warning('scenario type should be MultiHopScenario')
         reference_context = self.make_contexts(scenario)
         prompt_input = QueryConditions(
             persona=scenario.persona,

diff --git a/src/ragas/testset/synthesizers/single_hop/base.py b/src/ragas/testset/synthesizers/single_hop/base.py
@@ -119,9 +119,10 @@ def convert_to_scenario(self, data: t.Dict[str, t.Any]) -> SingleHopScenario:
         )
 
     async def _generate_sample(
-        self, scenario: SingleHopScenario, callbacks: Callbacks
+        self, scenario: Scenario, callbacks: Callbacks
     ) -> SingleTurnSample:
-
+        if not isinstance(scenario, SingleHopScenario):
+            logger.warning('scenario type should be SingleHopScenario')
         reference_context = scenario.nodes[0].properties.get("page_content", "")
         prompt_input = QueryCondition(
             persona=scenario.persona,