Skip to content

Commit d6863a0

Browse files
committed
feat(parallel): Make base_runner the main argument
1 parent 96536dd commit d6863a0

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/anemoi/inference/runners/parallel.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@
2424
from anemoi.inference.config import Configuration
2525
from anemoi.inference.output import Output
2626

27+
from ..decorators import main_argument
2728
from ..outputs import create_output
2829
from ..runner import Runner
2930
from ..runners import create_runner
@@ -48,6 +49,7 @@ def create_parallel_runner(config: Configuration, pid: int) -> None:
4849

4950

5051
@runner_registry.register("parallel")
52+
@main_argument("base_runner")
5153
class ParallelRunnerFactory:
5254
"""Creates a ParallelRunner with a dynamic base class."""
5355

0 commit comments

Comments
 (0)