Commit 62818e7c authored by Samuel GAIST's avatar Samuel GAIST
Browse files

[execution][local] Properly handle failure on setup and prepare

parent 78edabbd
......@@ -284,7 +284,12 @@ class LocalExecutor(BaseExecutor):
self.runner = self.algorithm.runner()
retval = self.runner.setup(['parameters'])
retval = self.runner.prepare(self.data_loaders)
if not retval:
raise RuntimeError("Algorithm setup failed")
prepared = self.runner.prepare(self.data_loaders)
if not prepared:
raise RuntimeError("Algorithm prepare failed")
if not self.input_list or not self.output_list:
raise RuntimeError("I/O for execution block has not yet been set up")
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment