Skip to content

Commit 58aefb2

Browse files
committed
Force specific instance specification in task generation
1 parent b520ca5 commit 58aefb2

1 file changed

Lines changed: 6 additions & 4 deletions

File tree

src/browsergym/workarena/tasks/scripts/knowledge.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
# TODO: Can we delete this file?
12
import json
23
import random
34

@@ -9,9 +10,8 @@
910
from tqdm import tqdm
1011

1112

12-
def generate_all_kb_configs(instance=None, num_configs=1000) -> list[dict]:
13+
def generate_all_kb_configs(instance, num_configs=1000) -> list[dict]:
1314
"""Generate all possible KB configs"""
14-
instance = instance if instance is not None else SNowInstance()
1515
with open(KB_FILEPATH, "r") as f:
1616
kb_entries = json.load(f)
1717
all_configs = []
@@ -33,5 +33,7 @@ def generate_all_kb_configs(instance=None, num_configs=1000) -> list[dict]:
3333

3434

3535
if __name__ == "__main__":
36-
37-
validate_kb_configs()
36+
raise NotImplementedError(
37+
"Make sure to specific instance URL and credentials below, then comment this line."
38+
)
39+
generate_all_kb_configs(instance=SNowInstance(snow_url=None))

0 commit comments

Comments
 (0)