dilithjay commited on
Commit
a89b932
·
1 Parent(s): 90ca6f7

Fix error when no dataset selected

Browse files
Files changed (1) hide show
  1. app.py +11 -6
app.py CHANGED
@@ -58,15 +58,22 @@ def get_tree(exp_num: int = 1, tree_num: int = 1):
58
 
59
 
60
  def change_expression(exp_num: int = 1, tree_num: int = 1):
61
- paths = glob(os.path.join(global_logs_dir, f"tree_{int(exp_num)}_*.txt"))
 
 
 
 
62
  tree_num = max(min(len(paths), tree_num), 1)
63
 
64
  tree = get_tree(exp_num, tree_num)
65
 
66
- return tree, gr.Slider.update(value=tree_num, maximum=len(paths))
67
 
68
 
69
  def find_expression(dataset_name: str):
 
 
 
70
  global global_logs_dir
71
  meta_data_df = pd.read_csv(meta_data_file, index_col="Name")
72
  n_channels = meta_data_df["Channels"][dataset_name]
@@ -135,9 +142,7 @@ def find_expression(dataset_name: str):
135
  )
136
  )
137
 
138
- yield tree, top_5_str, gr.Slider.update(
139
- value=i, maximum=i, interactive=True
140
- ), gr.Slider.update(value=1, maximum=len(data[-1][0]), interactive=True)
141
 
142
 
143
  with gr.Blocks(title="IndexRL") as demo:
@@ -176,7 +181,7 @@ with gr.Blocks(title="IndexRL") as demo:
176
  find_exp_event = find_exp_btn.click(
177
  find_expression,
178
  inputs=[select_dataset],
179
- outputs=[out_exp_tree, best_exps, select_exp, select_tree],
180
  )
181
  stop_btn.click(fn=None, inputs=None, outputs=None, cancels=[find_exp_event])
182
  select_exp.change(
 
58
 
59
 
60
  def change_expression(exp_num: int = 1, tree_num: int = 1):
61
+ try:
62
+ paths = glob(os.path.join(global_logs_dir, f"tree_{int(exp_num)}_*.txt"))
63
+ except TypeError:
64
+ return "", gr.Slider.update()
65
+
66
  tree_num = max(min(len(paths), tree_num), 1)
67
 
68
  tree = get_tree(exp_num, tree_num)
69
 
70
+ return tree, gr.Slider.update(value=tree_num, maximum=len(paths), interactive=True)
71
 
72
 
73
  def find_expression(dataset_name: str):
74
+ if dataset_name == "":
75
+ return ("", gr.Slider.update(value=1, interactive=False))
76
+
77
  global global_logs_dir
78
  meta_data_df = pd.read_csv(meta_data_file, index_col="Name")
79
  n_channels = meta_data_df["Channels"][dataset_name]
 
142
  )
143
  )
144
 
145
+ yield top_5_str, gr.Slider.update(value=i, maximum=i, interactive=True)
 
 
146
 
147
 
148
  with gr.Blocks(title="IndexRL") as demo:
 
181
  find_exp_event = find_exp_btn.click(
182
  find_expression,
183
  inputs=[select_dataset],
184
+ outputs=[best_exps, select_exp],
185
  )
186
  stop_btn.click(fn=None, inputs=None, outputs=None, cancels=[find_exp_event])
187
  select_exp.change(