fixed file import error in tests/test_domain_builder.py

AI-Planning · Dec 22, 2024 · 558e975 · 558e975
1 parent 05d15ee
commit 558e975
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/tests/test_domain_builder.py b/tests/test_domain_builder.py
@@ -9,9 +9,9 @@ def setUp(self):
         self.domain_builder = DomainBuilder()
 
     def test_extract_type(self):
-        mock_llm_1 = MockLLM([load_file("tests/test_prompts/test_extract_type/01.txt")])
-        mock_llm_2 = MockLLM([load_file("tests/test_prompts/test_extract_type/02.txt")])
-        mock_llm_3 = MockLLM([load_file("tests/test_prompts/test_extract_type/03.txt")])
+        mock_llm_1 = MockLLM([load_file("tests/test_prompts/test_domain_builder/test_extract_type/01.txt")])
+        mock_llm_2 = MockLLM([load_file("tests/test_prompts/test_domain_builder/test_extract_type/02.txt")])
+        mock_llm_3 = MockLLM([load_file("tests/test_prompts/test_domain_builder/test_extract_type/03.txt")])
 
         types, _ = self.domain_builder.extract_type(
             model=mock_llm_1,
@@ -49,13 +49,13 @@ def test_extract_type(self):
 
     def test_extract_type_hierarchy(self):
         mock_llm_1 = MockLLM(
-            [load_file("tests/test_prompts/test_extract_type_hierarchy/01.txt")]
+            [load_file("tests/test_prompts/test_domain_builder/test_extract_type_hierarchy/01.txt")]
         )
         mock_llm_2 = MockLLM(
-            [load_file("tests/test_prompts/test_extract_type_hierarchy/02.txt")]
+            [load_file("tests/test_prompts/test_domain_builder/test_extract_type_hierarchy/02.txt")]
         )
         mock_llm_3 = MockLLM(
-            [load_file("tests/test_prompts/test_extract_type_hierarchy/03.txt")]
+            [load_file("tests/test_prompts/test_domain_builder/test_extract_type_hierarchy/03.txt")]
         )
 
         expected_hierarchy = {