Skip to content

Commit ea73385

Browse files
committed
Clean up RunSet.__repr__, add config file
1 parent af05639 commit ea73385

File tree

1 file changed

+13
-17
lines changed

1 file changed

+13
-17
lines changed

cmdstanpy/stanfit/runset.py

Lines changed: 13 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -102,25 +102,21 @@ def __init__(
102102
]
103103

104104
def __repr__(self) -> str:
105-
repr = 'RunSet: chains={}, chain_ids={}, num_processes={}'.format(
106-
self._chains, self._chain_ids, self._num_procs
107-
)
108-
repr = '{}\n cmd (chain 1):\n\t{}'.format(repr, self.cmd(0))
109-
repr = '{}\n retcodes={}'.format(repr, self._retcodes)
110-
repr = f'{repr}\n per-chain output files (showing chain 1 only):'
111-
repr = '{}\n csv_file:\n\t{}'.format(repr, self._csv_files[0])
105+
lines = [
106+
f"RunSet: chains={self._chains}, chain_ids={self._chain_ids}, "
107+
f"num_processes={self._num_procs}",
108+
f" cmd (chain 1):\n\t{self.cmd(0)}",
109+
f" retcodes={self._retcodes}",
110+
" per-chain output files (showing chain 1 only):",
111+
f" csv_file:\n\t{self._csv_files[0] if self._csv_files else ''}",
112+
]
112113
if self._args.save_latent_dynamics:
113-
repr = '{}\n diagnostics_file:\n\t{}'.format(
114-
repr, self._diagnostic_files[0]
115-
)
114+
lines.append(f" diagnostics_file:\n\t{self._diagnostic_files[0]}")
116115
if self._args.save_profile:
117-
repr = '{}\n profile_file:\n\t{}'.format(
118-
repr, self._profile_files[0]
119-
)
120-
repr = '{}\n console_msgs (if any):\n\t{}'.format(
121-
repr, self._stdout_files[0]
122-
)
123-
return repr
116+
lines.append(f" profile_file:\n\t{self._profile_files[0]}")
117+
lines.append(f" console_msgs (if any):\n\t{self._stdout_files[0]}")
118+
lines.append(f" config_files:\n\t{self._config_files[0]}")
119+
return '\n'.join(lines)
124120

125121
@property
126122
def model(self) -> str:

0 commit comments

Comments
 (0)