def METHOD_NAME(self, dataset: Dataset) -> Dataset:
    from hail.context import TemporaryFilename
    tempfile = TemporaryFilename(prefix=f'persist_{type(dataset).__name__}')
    persisted = dataset.checkpoint(tempfile.__enter__())
    self._persisted_locations[persisted] = (tempfile, dataset)
    return persisted