Fix pyproject to allow local install and dynamic requirements

pyproject.toml

@@ -3,7 +3,7 @@ requires = ["setuptools >= 61.0.0"]
 build-backend = "setuptools.build_meta"
 
 [project]
-name = "tree-of-thought-llm"
+name = "tree_of_thought_llm"
 version = "0.1.0"
 description = 'Official Implementation of "Tree of Thoughts: Deliberate Problem Solving with Large Language Models"'
 readme = "README.md"
@@ -27,5 +27,8 @@ dynamic=["dependencies"]
 [tool.setuptools.dynamic]
 dependencies = {file = ["requirements.txt"]}
 
+[tool.setuptools.packages.find]
+where = ["tree_of_thought_llm"]  # list of folders that contain the packages (["."] by default)
+
 [project.urls]
 Homepage = "https://github.com/princeton-nlp/tree-of-thought-llm"

run.py

@@ -5,7 +5,7 @@ import argparse
 import numpy as np
 from functools import partial
 from models import gpt, gpt_usage
-from tasks import get_task
+from tree_of_thought_llm.tasks import get_task
 
 def get_value(task, x, y, n_evaluate_sample, cache_value=True):
     value_prompt = task.value_prompt_wrap(x, y)
@@ -100,9 +100,9 @@ def run(args):
     global gpt
     gpt = partial(gpt, model=args.backend, temperature=args.temperature)
     if args.naive_run:
-        file = f'logs/{args.task}/{args.backend}_{args.temperature}_naive_{args.prompt_sample}_sample_{args.n_generate_sample}_start{args.task_start_index}_end{args.task_end_index}.json'
+        file = f'tree_of_thought_llm/logs/{args.task}/{args.backend}_{args.temperature}_naive_{args.prompt_sample}_sample_{args.n_generate_sample}_start{args.task_start_index}_end{args.task_end_index}.json'
     else:
-        file = f'logs/{args.task}/{args.backend}_{args.temperature}_{args.method_generate}{args.n_generate_sample}_{args.method_evaluate}{args.n_evaluate_sample}_{args.method_select}{args.n_select_sample}_start{args.task_start_index}_end{args.task_end_index}.json'
+        file = f'tree_of_thought_llm/logs/{args.task}/{args.backend}_{args.temperature}_{args.method_generate}{args.n_generate_sample}_{args.method_evaluate}{args.n_evaluate_sample}_{args.method_select}{args.n_select_sample}_start{args.task_start_index}_end{args.task_end_index}.json'
     os.makedirs(os.path.dirname(file), exist_ok=True)
 
     for i in range(args.task_start_index, args.task_end_index):

tree_of_thought_llm/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.1.0"

tree_of_thought_llm/tasks/crosswords.py

@@ -1,7 +1,6 @@
 import re
 import json
-import os
-from tasks.base import Task, DATA_PATH
+from tasks.base import Task
 from prompts.crosswords import * 
 from models import gpt