Skip to content

Commit cf17ac5

Browse files
authored
Fix LSQB benchmark runs (#5820)
* fix lsqb benchmark serialization after single file database change
1 parent 6f59330 commit cf17ac5

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

benchmark/lsqb/benchmark_runner.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -126,7 +126,7 @@ def run_query(conn, query_spec):
126126

127127

128128
def run_kuzu(sf, serialized_graph_path, num_threads):
129-
db = kuzu.Database(serialized_graph_path)
129+
db = kuzu.Database(os.path.join(serialized_graph_path, "db.kuzu"))
130130
conn = kuzu.Connection(db, num_threads=num_threads)
131131
if timeout is not None:
132132
conn.set_query_timeout(timeout)

benchmark/lsqb/serializer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -144,7 +144,7 @@ def serialize(dataset_name, dataset_path, serialized_graph_path):
144144
shutil.rmtree(serialized_graph_path, ignore_errors=True)
145145
os.mkdir(serialized_graph_path)
146146

147-
db = kuzu.Database(serialized_graph_path)
147+
db = kuzu.Database(os.path.join(serialized_graph_path, "db.kuzu"))
148148
conn = kuzu.Connection(db)
149149
logging.info("Successfully connected")
150150

0 commit comments

Comments
 (0)