"vscode:/vscode.git/clone" did not exist on "614a7b1bb09acdd9ad8b856465d12ae82643bd76"
Commit 6833b906 authored by Jonas Kaufmann's avatar Jonas Kaufmann Committed by Antoine Kaufmann
Browse files

experiment runners: fix experiment not passed in super call

parent 6d80b901
...@@ -316,7 +316,7 @@ class ExperimentBaseRunner(ABC): ...@@ -316,7 +316,7 @@ class ExperimentBaseRunner(ABC):
class ExperimentSimpleRunner(ExperimentBaseRunner): class ExperimentSimpleRunner(ExperimentBaseRunner):
"""Simple experiment runner with just one executor.""" """Simple experiment runner with just one executor."""
def __init__(self, executor: Executor, exp: Experiment, *args, **kwargs): def __init__(self, executor: Executor, *args, **kwargs):
self.executor = executor self.executor = executor
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
...@@ -329,7 +329,7 @@ class ExperimentDistributedRunner(ExperimentBaseRunner): ...@@ -329,7 +329,7 @@ class ExperimentDistributedRunner(ExperimentBaseRunner):
def __init__(self, execs, exp: DistributedExperiment, *args, **kwargs): def __init__(self, execs, exp: DistributedExperiment, *args, **kwargs):
self.execs = execs self.execs = execs
super().__init__(*args, **kwargs) super().__init__(exp, *args, **kwargs)
self.exp = exp # overrides the type in the base class self.exp = exp # overrides the type in the base class
assert self.exp.num_hosts <= len(execs) assert self.exp.num_hosts <= len(execs)
......
Markdown is supported
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