Reformat long strings in contexts.py

这个提交包含在:
nmannall
2024-08-16 15:32:34 +01:00
父节点 c288e8eb7c
当前提交 93dadd9f1e

查看文件

@@ -152,7 +152,7 @@ class Context:
def print_sim_time_taken(self) -> None: def print_sim_time_taken(self) -> None:
"""Prints the total simulation time based on context.""" """Prints the total simulation time based on context."""
s = ( s = (
f"=== Simulation completed in " "=== Simulation completed in "
f"{humanize.precisedelta(datetime.timedelta(seconds=self.sim_end_time - self.sim_start_time), format='%0.4f')}" f"{humanize.precisedelta(datetime.timedelta(seconds=self.sim_end_time - self.sim_start_time), format='%0.4f')}"
) )
logger.basic(f"{s} {'=' * (get_terminal_width() - 1 - len(s))}\n") logger.basic(f"{s} {'=' * (get_terminal_width() - 1 - len(s))}\n")
@@ -169,20 +169,16 @@ class MPIContext(Context):
requested_mpi_size = np.product(config.sim_config.mpi) requested_mpi_size = np.product(config.sim_config.mpi)
if self.comm.size < requested_mpi_size: if self.comm.size < requested_mpi_size:
logger.error( logger.error(
( f"MPI_COMM_WORLD size of {self.comm.size} is too small for requested dimensions of"
f"MPI_COMM_WORLD size of {self.comm.size} is too small for requested dimensions of" f" {config.sim_config.mpi}. {requested_mpi_size} ranks are required."
f" {config.sim_config.mpi}. {requested_mpi_size} ranks are required."
)
) )
raise ValueError raise ValueError
if self.rank >= requested_mpi_size: if self.rank >= requested_mpi_size:
logger.warn( logger.warn(
( f"Rank {self.rank}: Only {requested_mpi_size} MPI ranks required for the"
f"Rank {self.rank}: Only {requested_mpi_size} MPI ranks required for the" " dimensions specified. Either increase your MPI dimension size, or request"
" dimensions specified. Either increase your MPI dimension size, or request" " fewer MPI tasks."
" fewer MPI tasks."
)
) )
exit() exit()