fix: fix name typos!

TogetherCrew · Nov 27, 2024 · 737b063 · 737b063
1 parent f9dff40
commit 737b063
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/unit/test_website_query_engine.py b/tests/unit/test_website_query_engine.py
@@ -9,9 +9,9 @@
 class TestNotionQueryEngine(TestCase):
     def setUp(self) -> None:
         community_id = "sample_community"
-        self.notion_query_engine = WebsiteQueryEngine(community_id)
+        self.website_query_engine = WebsiteQueryEngine(community_id)
 
     def test_prepare_engine(self):
-        notion_query_engine = self.notion_query_engine.prepare(testing=True)
-        print(notion_query_engine.__dict__)
-        self.assertIsInstance(notion_query_engine.retriever, VectorIndexRetriever)
+        website_query_engine = self.website_query_engine.prepare(testing=True)
+        print(website_query_engine.__dict__)
+        self.assertIsInstance(website_query_engine.retriever, VectorIndexRetriever)