From 6e87f97aeb59cf72be87eca903a4afe028896ddf Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?The=20Vinh=20LUONG=20=28L=C6=AF=C6=A0NG=20Th=E1=BA=BF=20Vi?= =?UTF-8?q?nh=29?= Date: Sat, 5 Oct 2024 16:06:24 -0700 Subject: [PATCH] refactor examples/semiconductor-etching-yieldguard/dana module --- examples/semiconductor-etching-yieldguard/dana.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/examples/semiconductor-etching-yieldguard/dana.py b/examples/semiconductor-etching-yieldguard/dana.py index 3f596558f..d7d82fce0 100644 --- a/examples/semiconductor-etching-yieldguard/dana.py +++ b/examples/semiconductor-etching-yieldguard/dana.py @@ -20,10 +20,11 @@ load_dotenv() -DATA_DIR_PATH: Path = Path(__file__).parent / 'data' +BASE_DIR: Path = Path(__file__).parent +DATA_DIR_PATH: Path = BASE_DIR / 'data' -EXPERT_KNOWLEDGE_FILE_PATH: Path = Path(__file__).parent / 'expertise' / 'expert-knowledge.txt' +EXPERT_KNOWLEDGE_FILE_PATH: Path = BASE_DIR / 'expertise' / 'expert-knowledge.txt' with open(file=EXPERT_KNOWLEDGE_FILE_PATH, buffering=-1, encoding='utf-8', @@ -33,8 +34,7 @@ opener=None) as f: EXPERT_KNOWLEDGE: str = f.read() - -EXPERT_PROGRAMS_FILE_PATH: Path = Path(__file__).parent / 'expertise' / 'expert-programs.yml' +EXPERT_PROGRAMS_FILE_PATH: Path = BASE_DIR / 'expertise' / 'expert-programs.yml' with open(file=EXPERT_PROGRAMS_FILE_PATH, buffering=-1, encoding='utf-8', @@ -58,7 +58,7 @@ def get_or_create_dana(use_semikong_lm: bool = False, max_depth=2, max_subtasks_ return DANA(knowledge={EXPERT_KNOWLEDGE}, program_store=program_store, programmer=HTPlanner(lm=lm, max_depth=max_depth, max_subtasks_per_decomp=max_subtasks_per_decomp), - resources={FileResource(path=DATA_DIR_PATH, lm=lm)}) + resources={FileResource(path=DATA_DIR_PATH, re_index=True, lm=lm)}) if __name__ == '__main__':