Spaces:
Sleeping
Sleeping
ping98k
commited on
Commit
·
a959b3c
1
Parent(s):
aebdef3
fix error
Browse files- main.py +3 -3
- tournament_utils.py +1 -1
main.py
CHANGED
|
@@ -6,10 +6,10 @@ from tqdm import tqdm
|
|
| 6 |
import matplotlib.pyplot as plt
|
| 7 |
from tournament_utils import generate_players, prompt_score, prompt_play
|
| 8 |
|
| 9 |
-
NUM_TOP_PICKS_DEFAULT = int(os.getenv("NUM_TOP_PICKS",
|
| 10 |
-
POOL_SIZE_DEFAULT = int(os.getenv("POOL_SIZE",
|
| 11 |
MAX_WORKERS_DEFAULT = int(os.getenv("MAX_WORKERS", 10))
|
| 12 |
-
NUM_GENERATIONS_DEFAULT = int(os.getenv("NUM_GENERATIONS",
|
| 13 |
API_BASE_DEFAULT = os.getenv("OPENAI_API_BASE", "")
|
| 14 |
API_TOKEN_DEFAULT = os.getenv("OPENAI_API_KEY", "")
|
| 15 |
SCORE_FILTER_DEFAULT = os.getenv("ENABLE_SCORE_FILTER", "true").lower() == "true"
|
|
|
|
| 6 |
import matplotlib.pyplot as plt
|
| 7 |
from tournament_utils import generate_players, prompt_score, prompt_play
|
| 8 |
|
| 9 |
+
NUM_TOP_PICKS_DEFAULT = int(os.getenv("NUM_TOP_PICKS", 3))
|
| 10 |
+
POOL_SIZE_DEFAULT = int(os.getenv("POOL_SIZE", 5))
|
| 11 |
MAX_WORKERS_DEFAULT = int(os.getenv("MAX_WORKERS", 10))
|
| 12 |
+
NUM_GENERATIONS_DEFAULT = int(os.getenv("NUM_GENERATIONS", 10))
|
| 13 |
API_BASE_DEFAULT = os.getenv("OPENAI_API_BASE", "")
|
| 14 |
API_TOKEN_DEFAULT = os.getenv("OPENAI_API_KEY", "")
|
| 15 |
SCORE_FILTER_DEFAULT = os.getenv("ENABLE_SCORE_FILTER", "true").lower() == "true"
|
tournament_utils.py
CHANGED
|
@@ -23,7 +23,7 @@ def prompt_score(
|
|
| 23 |
prompt = f"""Evaluate the output below on the following criteria:
|
| 24 |
{criteria_block}
|
| 25 |
|
| 26 |
-
Return JSON exactly like: {{"
|
| 27 |
|
| 28 |
Instruction:
|
| 29 |
{instruction}
|
|
|
|
| 23 |
prompt = f"""Evaluate the output below on the following criteria:
|
| 24 |
{criteria_block}
|
| 25 |
|
| 26 |
+
Return JSON exactly like: {{"scores": [{example_scores}]}}.
|
| 27 |
|
| 28 |
Instruction:
|
| 29 |
{instruction}
|