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):
class ExperimentSimpleRunner(ExperimentBaseRunner):
"""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
super().__init__(*args, **kwargs)
......@@ -329,7 +329,7 @@ class ExperimentDistributedRunner(ExperimentBaseRunner):
def __init__(self, execs, exp: DistributedExperiment, *args, **kwargs):
self.execs = execs
super().__init__(*args, **kwargs)
super().__init__(exp, *args, **kwargs)
self.exp = exp # overrides the type in the base class
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