@@ -216,7 +216,8 @@ def gen_sql_from_stream(query_stream_file_path):
216216
217217 parsed = parse_query_content (q )
218218
219- def add_to_extended_queries (query_type , idx , parsed ):
219+ def add_to_extended_queries (query_type , i , parsed ):
220+ idx = i + 1
220221 subquery_cnt = len (parsed [query_type ])
221222 if query_type == 'benchmark' :
222223 dict_key = f"{ query_name } _part{ idx } " if subquery_cnt > 1 else query_name
@@ -227,11 +228,11 @@ def add_to_extended_queries(query_type, idx, parsed):
227228 extended_queries [dict_key ] = query_part
228229
229230 for i in range (len (parsed ['setup' ])):
230- add_to_extended_queries ('setup' , i + 1 , parsed )
231+ add_to_extended_queries ('setup' , i , parsed )
231232 for i in range (len (parsed ['benchmark' ])):
232- add_to_extended_queries ('benchmark' , i + 1 , parsed )
233+ add_to_extended_queries ('benchmark' , i , parsed )
233234 for i in range (len (parsed ['cleanup' ])):
234- add_to_extended_queries ('cleanup' , i + 1 , parsed )
235+ add_to_extended_queries ('cleanup' , i , parsed )
235236
236237 # add "-- start" string back to each query
237238 for q_name , q_content in extended_queries .items ():
0 commit comments