Security: remove API keys
Browse files
experiments/run_paper_scale_v5.py
CHANGED
|
@@ -15,19 +15,19 @@ Fixes over V4:
|
|
| 15 |
import os, sys, json, time, csv, random, copy
|
| 16 |
from collections import defaultdict
|
| 17 |
|
| 18 |
-
sys.path.insert(0, '/root/autodl-tmp/
|
| 19 |
-
sys.path.insert(0, '/root/autodl-tmp/
|
| 20 |
|
| 21 |
from openai import OpenAI
|
| 22 |
from baselines.evorm_wrappers.matchgpt import run_matchgpt, load_er_dataset
|
| 23 |
from baselines.evorm_wrappers.lela_el import run_lela, run_schema_matching
|
| 24 |
|
| 25 |
# Config
|
| 26 |
-
API_BASE = "https://
|
| 27 |
-
API_KEY
|
| 28 |
-
DATA_DIR = '/root/autodl-tmp/
|
| 29 |
-
PERSIST_DIR = '/root/autodl-tmp/
|
| 30 |
-
RESULTS_DIR = '/root/autodl-tmp/
|
| 31 |
|
| 32 |
EVAL_N = 300
|
| 33 |
WARMUP_N = 100
|
|
|
|
| 15 |
import os, sys, json, time, csv, random, copy
|
| 16 |
from collections import defaultdict
|
| 17 |
|
| 18 |
+
sys.path.insert(0, '/root/autodl-tmp/EvoRM')
|
| 19 |
+
sys.path.insert(0, '/root/autodl-tmp/EvoRM/baselines/evorm_wrappers')
|
| 20 |
|
| 21 |
from openai import OpenAI
|
| 22 |
from baselines.evorm_wrappers.matchgpt import run_matchgpt, load_er_dataset
|
| 23 |
from baselines.evorm_wrappers.lela_el import run_lela, run_schema_matching
|
| 24 |
|
| 25 |
# Config
|
| 26 |
+
API_BASE = os.environ.get("OPENAI_BASE_URL", "https://api.openai.com/v1")
|
| 27 |
+
# API_KEY set via environment variable
|
| 28 |
+
DATA_DIR = '/root/autodl-tmp/EvoRM/data'
|
| 29 |
+
PERSIST_DIR = '/root/autodl-tmp/EvoRM/persist_v5'
|
| 30 |
+
RESULTS_DIR = '/root/autodl-tmp/EvoRM/results'
|
| 31 |
|
| 32 |
EVAL_N = 300
|
| 33 |
WARMUP_N = 100
|