Skip to content

Commit af05639

Browse files
committed
Add config files to RunSet
1 parent 097d4dc commit af05639

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

cmdstanpy/stanfit/runset.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -56,20 +56,29 @@ def __init__(
5656
)
5757
self._stdout_files, self._profile_files = [], []
5858
self._csv_files, self._diagnostic_files = [], []
59+
self._config_files = []
5960

6061
# per-process output files
6162
if one_process_per_chain and chains > 1:
6263
self._stdout_files = [
6364
self.gen_file_name(".txt", extra="stdout", id=id)
6465
for id in self._chain_ids
6566
]
67+
self._config_files = [
68+
os.path.join(
69+
self._outdir, f"{self._base_outfile}_{id}_config.json"
70+
)
71+
for id in self._chain_ids
72+
]
73+
6674
if args.save_profile:
6775
self._profile_files = [
6876
self.gen_file_name(".csv", extra="profile", id=id)
6977
for id in self._chain_ids
7078
]
7179
else:
7280
self._stdout_files = [self.gen_file_name(".txt", extra="stdout")]
81+
self._config_files = [self.gen_file_name(".json", extra="config")]
7382
if args.save_profile:
7483
self._profile_files = [
7584
self.gen_file_name(".csv", extra="profile")
@@ -196,6 +205,13 @@ def stdout_files(self) -> list[str]:
196205
"""
197206
return self._stdout_files
198207

208+
@property
209+
def config_files(self) -> list[str]:
210+
"""
211+
List of paths to CmdStan config json files.
212+
"""
213+
return self._config_files
214+
199215
def _check_retcodes(self) -> bool:
200216
"""Returns ``True`` when all chains have retcode 0."""
201217
return all(retcode == 0 for retcode in self._retcodes)

0 commit comments

Comments
 (0)