Skip to content

Commit a62e7d2

Browse files
committed
Simplify RunSet.gen_file_name
1 parent 5429dc4 commit a62e7d2

File tree

1 file changed

+6
-11
lines changed

1 file changed

+6
-11
lines changed

cmdstanpy/stanfit/runset.py

Lines changed: 6 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -214,17 +214,12 @@ def gen_file_name(
214214
self, suffix: str, *, extra: str = "", id: int | None = None
215215
) -> str:
216216
"""Generate a standard file name according to CmdStan output pattern"""
217-
match (id, extra):
218-
case (None, ""):
219-
file = f"{self._base_outfile}{suffix}"
220-
case (None, extra) if extra != "":
221-
file = f"{self._base_outfile}_{extra}{suffix}"
222-
case (id, ""):
223-
file = f"{self._base_outfile}_{id}{suffix}"
224-
case (id, extra) if extra != "":
225-
file = f"{self._base_outfile}_{extra}_{id}{suffix}"
226-
case _:
227-
raise ValueError("Cannot construct valid file name")
217+
file = self._base_outfile
218+
if extra:
219+
file += f"_{extra}"
220+
if id is not None:
221+
file += f"_{id}"
222+
file += suffix
228223
return os.path.join(self._outdir, file)
229224

230225
def _retcode(self, idx: int) -> int:

0 commit comments

Comments
 (0)