Liu Yiwen commited on
Commit
a2f1673
·
1 Parent(s): 0d31272

增加自动获取split功能

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -309,7 +309,9 @@ with gr.Blocks() as demo:
309
  question_info = get_question_info(df)
310
  ret[qusetion_id_box] = gr.update(value = df[COLUMN_ID][0])
311
 
312
- lotsa_config, lotsa_split, lotsa_page = str(df[COLUMN_SOURCE][0]).split('/')[-1], 'train', eval(df[COLUMN_TS_ID][0])
 
 
313
  start_index, end_index = df[COLUMN_START_INDEX][0], df[COLUMN_END_INDEX][0]
314
  interval = None if np.isnan(start_index) or np.isnan(end_index) else [start_index, end_index]
315
  lotsa_subtargets = eval(df[COLUMN_TARGET_ID][0])
 
309
  question_info = get_question_info(df)
310
  ret[qusetion_id_box] = gr.update(value = df[COLUMN_ID][0])
311
 
312
+ lotsa_config, lotsa_page = str(df[COLUMN_SOURCE][0]).split('/')[-1], eval(df[COLUMN_TS_ID][0])
313
+ #TODO: 对partial-train的处理
314
+ lotsa_split = get_parquet_splits(TARGET_DATASET, lotsa_config)[0]
315
  start_index, end_index = df[COLUMN_START_INDEX][0], df[COLUMN_END_INDEX][0]
316
  interval = None if np.isnan(start_index) or np.isnan(end_index) else [start_index, end_index]
317
  lotsa_subtargets = eval(df[COLUMN_TARGET_ID][0])