const tfrecord = require('tfrecord');

async function writeDemo() {
  const writer = await tfrecord.createWriter('data.tfrecord');

  const example = tfrecord.Example.fromObject({
    features: {
      feature: {
        answer: {  // The feature name.
          int64List: {
            value: [42],  // The feature value.
          },
        },
      },
    },
  });

  await writer.writeExample(example);
  await writer.close();
}

async function readDemo() {
  const reader = await tfrecord.createReader('data.tfrecord');
  let example;
  while (example = await reader.readExample()) {
    console.log('%j', example.toJSON());
  }
  // The reader auto-closes after it reaches the end of the file.
}

async function demo() {
  await writeDemo();
  await readDemo();
}

demo();