ExplicitCheckpointKeeper

class ExplicitCheckpointKeeper(keep: Collection[int])

Bases: CheckpointKeeper

Keep checkpoints at explicit steps.

Attributes Summary

keep

Methods Summary

__call__(steps)

Iterate over the steps for which checkpoints should be kept.

Attributes Documentation

Parameters:

keep (Collection[int])

keep: Collection[int] = <dataclasses._MISSING_TYPE object>

Methods Documentation

__call__(steps: Sequence[int]) → Iterator[int]

Iterate over the steps for which checkpoints should be kept.

Parameters:

steps (Sequence[int]) – the sorted list of steps at which checkpoints were written.

Yields:

the steps for which checkpoints should be kept

Return type:

Iterator[int]

# noqa:DAR302