diff --git a/src/pystencils_autodiff/graph_datahandling.py b/src/pystencils_autodiff/graph_datahandling.py index 27fc13b7027863a805d277af8787af0a82803a2d..8768edc5c7a696bc0c63f032c35aaf56261c0669 100644 --- a/src/pystencils_autodiff/graph_datahandling.py +++ b/src/pystencils_autodiff/graph_datahandling.py @@ -124,7 +124,7 @@ class TimeloopRun: self.time_steps = time_steps def __str__(self): - return (f'Timeloop:' + return ('Timeloop:' + '\nPre:\n' + '\n '.join(str(f) for f in self.timeloop._pre_run_functions) + f'\n{self.time_steps} time steps:\n'