Apply suggestions from code review

Co-Authored-By: abduld <[email protected]>

Author: fixmie[bot]

reader/record.go

@@ -9,6 +9,7 @@ type ImageRecord struct {
 	Image      *types.RGBImage
 }
 
+// ImageSegmentationRecord ...
 type ImageSegmentationRecord struct {
 	ID         uint64
 	LabelIndex float32

reader/tfrecord.go

@@ -15,6 +15,7 @@ import (
 	protobuf "github.com/ubccr/terf/protobuf"
 )
 
+// TFRecordReader ...
 type TFRecordReader struct {
 	r io.ReadCloser
 	*terf.Reader
@@ -32,6 +33,7 @@ func NewTFRecordReader(path string) (*TFRecordReader, error) {
 	}, nil
 }
 
+// NextRecord ...
 func (r *TFRecordReader) NextRecord(ctx context.Context) (*protobuf.Example, error) {
 	nxt, err := r.Reader.Next()
 	if err != nil {
@@ -40,6 +42,7 @@ func (r *TFRecordReader) NextRecord(ctx context.Context) (*protobuf.Example, err
 	return nxt, nil
 }
 
+// Next ...
 func (r *TFRecordReader) Next(ctx context.Context) (*ImageRecord, error) {
 	nxt, err := r.Reader.Next()
 	if err != nil {

reader/tfrecord/features.go

@@ -5,6 +5,7 @@ import (
 	protobuf "github.com/ubccr/terf/protobuf"
 )
 
+// FeatureBool ...
 func FeatureBool(rec *protobuf.Example, key string) bool {
 	return FeatureInt(rec, key) == 1
 }
@@ -24,10 +25,12 @@ func FeatureInt64(rec *protobuf.Example, key string) int64 {
 	return val.Int64List.Value[0]
 }
 
+// FeatureInt ...
 func FeatureInt(rec *protobuf.Example, key string) int {
 	return int(FeatureInt64(rec, key))
 }
 
+// FeatureInt32 ...
 func FeatureInt32(rec *protobuf.Example, key string) int32 {
 	return int32(FeatureInt64(rec, key))
 }
@@ -52,6 +55,7 @@ func FeatureFloat32(rec *protobuf.Example, key string) float32 {
 	return val.FloatList.Value[0]
 }
 
+// FeatureBytes ...
 func FeatureBytes(rec *protobuf.Example, key string) []byte {
 	return terf.ExampleFeatureBytes(rec, key)
 }

vision/coco.go

@@ -150,6 +150,7 @@ func (d *CocoValidationTFRecord) Load(ctx context.Context) error {
 	return d.loadRecord(ctx)
 }
 
+// Next ...
 func (d *CocoValidationTFRecord) Next(ctx context.Context) (dldataset.LabeledData, error) {
 	rec, err := d.recordReader.NextRecord(ctx)
 	if err != nil {