def _register_info(self, server): """Write a TensorBoardInfo file and arrange for its cleanup. Args: server: The result of `self._make_server()`. """ server_url = urllib.parse.urlparse(server.get_url()) info = manager.TensorBoardInfo( version=version.VERSION, start_time=int(time.time()), port=server_url.port, pid=os.getpid(), path_prefix=self.flags.path_prefix, logdir=self.flags.logdir or self.flags.logdir_spec, db=self.flags.db, cache_key=self.cache_key, ) atexit.register(manager.remove_info_file) manager.write_info_file(info)
def _make_info(i=0): """Make a sample TensorBoardInfo object. Args: i: Seed; vary this value to produce slightly different outputs. Returns: A type-correct `TensorBoardInfo` object. """ return manager.TensorBoardInfo( version=version.VERSION, start_time=1548973541 + i, port=6060 + i, pid=76540 + i, path_prefix="/foo", logdir="~/my_data/", db="", cache_key="asdf", )