hiro/hopsflow
1
0
Fork 0
mirror of https://github.com/vale981/hopsflow synced 2025-03-04 16:31:38 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

allow a callable to decide when to save a snapshot

Browse source
This commit is contained in:
Valentin Boettcher 2022-12-09 15:06:46 -05:00
parent cd5f82d044
commit e9900520c4
No known key found for this signature in database
GPG key ID: E034E12B7AF56ACE
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
hopsflow/util.py
View file

@ -840,7 +840,7 @@ def ensemble_mean_online(
save: str,
function: Callable[..., np.ndarray],
i: Optional[int] = None,
every: Optional[int] = None,
every: Optional[Union[int, Callable[[int], bool]]] = None,
) -> Optional[EnsembleValue]:
path = Path(save)
@ -863,7 +863,9 @@ def ensemble_mean_online(
aggregate = WelfordAggregator(result, i)
aggregate.dump(str(path))
if every is not None and aggregate.n % every == 0:
if every is not None and (
aggregate.n % every == 0 if isintance(every, int) else every(aggregate.n)
):
path.with_stem(f"{path.stem}_{aggregate.n}")
aggregate.dump(str(path))