Skip to content

Commit e52ddde

Browse files
committed
flake fix
1 parent b89a62c commit e52ddde

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

cmdstanpy/stanfit/mcmc.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -311,8 +311,7 @@ def _validate_csv_files(self) -> Dict[str, Any]:
311311
save_warmup=self._save_warmup,
312312
thin=self._thin,
313313
)
314-
time_info = cast(Dict[str, float], dzero.get("time") or {})
315-
self._chain_time.append(time_info)
314+
self._chain_time.append(dzero['time']) # type: ignore
316315
if not self._is_fixed_param:
317316
self._divergences[i] = dzero['ct_divergences']
318317
self._max_treedepths[i] = dzero['ct_max_treedepth']
@@ -325,8 +324,7 @@ def _validate_csv_files(self) -> Dict[str, Any]:
325324
save_warmup=self._save_warmup,
326325
thin=self._thin,
327326
)
328-
time_info = cast(Dict[str, float], drest.get("time") or {})
329-
self._chain_time.append(time_info)
327+
self._chain_time.append(drest['time']) # type: ignore
330328
for key in dzero:
331329
# check args that matter for parsing, plus name, version
332330
if (

0 commit comments

Comments
 (0)