def write(self, tf_example):
        """Writres the ``tf.train.Example`` proto to disk.

        Args:
            tf_example: a ``tf.train.Example`` proto
        """
        self._idx += 1
        self._writers[self._idx % self._num_shards].write(
            tf_example.SerializeToString()
        )
Writres the ``tf.train.Example`` proto to disk.

        Args:
            tf_example: a ``tf.train.Example`` proto
        
fiftyone/utils/tf.py