Skip to content

Commit 43ce420

Browse files
Fixing a bug in reusing database and adding database name as an option
Summary: This diff fixes a bug that disables skipping database loading Reviewed By: YifanYuan3 Differential Revision: D76067803
1 parent 917a8f2 commit 43ce420

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

benchpress/config/jobs.yml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -480,7 +480,7 @@
480480
args:
481481
- 'run'
482482
- '--dataset-path /flash23/'
483-
- '--dataset-name bpc_t93586_s2_synthetic_5GB'
483+
- '--dataset-name {dataset_name}'
484484
- '--warehouse-dir /flash23/warehouse'
485485
- '--shuffle-dir /flash23/spark_local_dir'
486486
- '--ipv4 {ipv4}'
@@ -489,6 +489,7 @@
489489
vars:
490490
- 'ipv4=0'
491491
- 'sanity=0'
492+
- 'dataset_name=bpc_t93586_s2_synthetic_5GB'
492493
hooks:
493494
- hook: copymove
494495
options:

packages/spark_standalone/templates/runner.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ def download_dataset(args):
5252

5353
def install_database(args):
5454
metadata_dir = os.path.join(SPARK_DIR, "spark-2.4.5-bin-hadoop2.7", "metastore_db")
55-
database_dir = os.path.join(args.warehouse_dir, f"{args.dataset_name}.db")
55+
database_dir = os.path.join(args.warehouse_dir, f"{args.dataset_name.lower()}.db")
5656
if os.path.exists(metadata_dir) and os.path.exists(database_dir):
5757
print("Database already created; directly run test")
5858
return

0 commit comments

Comments
 (0)