Skip to content

Commit

Permalink
update examples/semiconductor-etching-yieldguard/dana module
Browse files Browse the repository at this point in the history
  • Loading branch information
TheVinhLuong102 committed Oct 5, 2024
1 parent 88896e9 commit 7aa1858
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions examples/semiconductor-etching-yieldguard/dana.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
from dotenv import load_dotenv
import yaml

from openssa import DANA, ProgramStore, HTP, HTPlanner, HuggingFaceLM
from openssa import DANA, ProgramStore, HTP, HTPlanner, FileResource, HuggingFaceLM

# pylint: disable=wrong-import-order
from semikong_lm import SemiKongLM
Expand All @@ -20,6 +20,9 @@
load_dotenv()


DATA_DIR_PATH: Path = Path(__file__).parent / 'data'


EXPERT_KNOWLEDGE_FILE_PATH: Path = Path(__file__).parent / 'expertise' / 'expert-knowledge.txt'
with open(file=EXPERT_KNOWLEDGE_FILE_PATH,
buffering=-1,
Expand Down Expand Up @@ -55,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={})
resources={FileResource(path=DATA_DIR_PATH, lm=lm)})


if __name__ == '__main__':
Expand Down

0 comments on commit 7aa1858

Please sign in to comment.