Skip to content

Commit e649d23

Browse files
committed
now getting queries from data instead of build
1 parent 731db40 commit e649d23

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

benchmark/job/cli.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -129,6 +129,7 @@
129129
"33b",
130130
"33c",
131131
]
132+
JOB_QUERIES_DNAME = "job-queries"
132133

133134

134135
@click.group(name="job")
@@ -174,7 +175,7 @@ def _download_job_data(dbgym_cfg: DBGymConfig) -> None:
174175

175176

176177
def _download_job_queries(dbgym_cfg: DBGymConfig) -> None:
177-
_download_and_untar_dir(dbgym_cfg, JOB_QUERIES_URL, "job.tgz", "job-queries")
178+
_download_and_untar_dir(dbgym_cfg, JOB_QUERIES_URL, "job.tgz", JOB_QUERIES_DNAME)
178179

179180

180181
def _download_and_untar_dir(
@@ -230,7 +231,8 @@ def _generate_job_workload(
230231
with open(real_dpath / "order.txt", "w") as f:
231232
for qname in query_names:
232233
sql_fpath = (
233-
dbgym_cfg.cur_symlinks_build_path(mkdir=True) / ("job-queries.link")
234+
dbgym_cfg.cur_symlinks_data_path(mkdir=True)
235+
/ (f"{JOB_QUERIES_DNAME}.link")
234236
).resolve() / f"{qname}.sql"
235237
assert (
236238
sql_fpath.exists()

scripts/run_protox_e2e_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -200,5 +200,5 @@ def run_e2e_for_benchmark(benchmark_name: str, intended_dbdata_hardware: str) ->
200200
# Set the config file so that we use resources that don't conflict with normal usage (e.g. a different workspace, different ports, etc.).
201201
os.environ["DBGYM_CONFIG_PATH"] = str(E2ETEST_DBGYM_CONFIG_FPATH)
202202

203-
# run_e2e_for_benchmark("tpch", intended_dbdata_hardware)
203+
run_e2e_for_benchmark("tpch", intended_dbdata_hardware)
204204
run_e2e_for_benchmark("job", intended_dbdata_hardware)

0 commit comments

Comments
 (0)