Expose method and class - so that we can use it from user code (particularly since checkpoint directory is autogenerated now