From bff1ec25d3c022ceac338097d4c12196b37bbd8e Mon Sep 17 00:00:00 2001 From: bowen xian Date: Tue, 23 Jul 2024 07:57:46 +0000 Subject: [PATCH] process None in list when demo --- rdagent/log/ui/web.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/rdagent/log/ui/web.py b/rdagent/log/ui/web.py index 3019afaa..606788d9 100644 --- a/rdagent/log/ui/web.py +++ b/rdagent/log/ui/web.py @@ -308,7 +308,7 @@ class QlibFactorExpWindow(StWindow): # factor tasks ftm_msg = deepcopy(msg) - ftm_msg.content = exp.sub_workspace_list + ftm_msg.content = [ws for ws in exp.sub_workspace_list if ws] ObjectsTabsWindow(self.container.expander('Factor Tasks'), inner_class=WorkspaceWindow, mapper=lambda x: x.target_task.factor_name, @@ -335,10 +335,10 @@ class QlibModelExpWindow(StWindow): # model tasks _msg = deepcopy(msg) - _msg.content = exp.sub_workspace_list + _msg.content = [ws for ws in exp.sub_workspace_list if ws] ObjectsTabsWindow(self.container.expander('Model Tasks'), inner_class=WorkspaceWindow, - mapper=lambda x: x.target_task.name if x else 'None', + mapper=lambda x: x.target_task.name, ).consume_msg(_msg) # result @@ -389,7 +389,9 @@ class QlibTraceWindow(StWindow): elif isinstance(msg.content, QlibModelExperiment): self.current_win = QlibModelExpWindow(self.container) elif isinstance(msg.content, list): - + msg.content = [m for m in msg.content if m] + if len(msg.content) == 0: + return if isinstance(msg.content[0], FactorTask): self.current_win = ObjectsTabsWindow(self.container.expander('Factor Tasks'), FactorTaskWindow, lambda x: x.factor_name) elif isinstance(msg.content[0], ModelTask):