ping98k commited on
Commit
a959b3c
·
1 Parent(s): aebdef3
Files changed (2) hide show
  1. main.py +3 -3
  2. 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", 5))
10
- POOL_SIZE_DEFAULT = int(os.getenv("POOL_SIZE", 10))
11
  MAX_WORKERS_DEFAULT = int(os.getenv("MAX_WORKERS", 10))
12
- NUM_GENERATIONS_DEFAULT = int(os.getenv("NUM_GENERATIONS", 20))
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: {{"score": [{example_scores}]}}.
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}