From 6de3f011d271b8dd42a89cc09cc4d77e1e866a50 Mon Sep 17 00:00:00 2001 From: lward Date: Thu, 7 Mar 2024 17:42:28 -0500 Subject: [PATCH] Do not rely on the class name for the task name --- demo_apps/mpi-tasks/sim.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/demo_apps/mpi-tasks/sim.py b/demo_apps/mpi-tasks/sim.py index 1ada824..155d7b7 100644 --- a/demo_apps/mpi-tasks/sim.py +++ b/demo_apps/mpi-tasks/sim.py @@ -1,12 +1,13 @@ from pathlib import Path -from colmena.models import ExecutableTask +from colmena.models.tasks import ExecutableTask class Simulation(ExecutableTask): def __init__(self, executable: Path): super().__init__(executable=[executable.absolute()], + name='simulation', mpi=True, mpi_command_string='mpirun -np {total_ranks}')