Fix error when no dataset selected
Browse files
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 |
-
|
|
|
|
|
|
|
|
|
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
|
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=[
|
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(
|