Apply suggestions from code review

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

Author: fixmie[bot]

reader/tfrecord.go

@@ -92,6 +92,7 @@ func (r *TFRecordReader) Next(ctx context.Context) (*ImageRecord, error) {
 	}, nil
 }
 
+// Close ...
 func (r *TFRecordReader) Close() error {
 	return r.r.Close()
 }

reader/tfrecord/features.go

@@ -32,6 +32,7 @@ func FeatureInt32(rec *protobuf.Example, key string) int32 {
 	return int32(FeatureInt64(rec, key))
 }
 
+// FeatureFloat64 ...
 func FeatureFloat64(rec *protobuf.Example, key string) float64 {
 	return float64(FeatureFloat32(rec, key))
 }
@@ -55,10 +56,12 @@ func FeatureBytes(rec *protobuf.Example, key string) []byte {
 	return terf.ExampleFeatureBytes(rec, key)
 }
 
+// FeatureString ...
 func FeatureString(rec *protobuf.Example, key string) string {
 	return string(FeatureBytes(rec, key))
 }
 
+// FeatureBytesSlice ...
 func FeatureBytesSlice(rec *protobuf.Example, key string) [][]byte {
 	// TODO: return error if key is not found?
 	f, ok := rec.Features.Feature[key]
@@ -88,6 +91,7 @@ func FeatureStringSlice(rec *protobuf.Example, key string) []string {
 	return res
 }
 
+// FeatureInt64Slice ...
 func FeatureInt64Slice(rec *protobuf.Example, key string) []int64 {
 
 	f, ok := rec.Features.Feature[key]
@@ -118,6 +122,7 @@ func FeatureIntSlice(rec *protobuf.Example, key string) []int {
 	return res
 }
 
+// FeatureInt32Slice ...
 func FeatureInt32Slice(rec *protobuf.Example, key string) []int32 {
 	slice := FeatureInt64Slice(rec, key)
 	if slice == nil {
@@ -132,6 +137,7 @@ func FeatureInt32Slice(rec *protobuf.Example, key string) []int32 {
 	return res
 }
 
+// FeatureFloat64Slice ...
 func FeatureFloat64Slice(rec *protobuf.Example, key string) []float64 {
 	slice := FeatureFloat32Slice(rec, key)
 	if slice == nil {

vision/cityscape_class.go

@@ -1,5 +1,6 @@
 package vision
 
+// CityScapeClass ...
 type CityScapeClass struct {
 	Name         string
 	ID           int64
@@ -11,6 +12,7 @@ type CityScapeClass struct {
 	Color        []uint8
 }
 
+// CityScapeClasses ...
 var CityScapeClasses []CityScapeClass
 
 func init() {

vision/coco.go

@@ -34,6 +34,7 @@ type CocoLabeledImage struct {
 	data     *types.RGBImage
 }
 
+// CocoValidationTFRecord ...
 type CocoValidationTFRecord struct {
 	base
 	name           string
@@ -68,17 +69,20 @@ func (d *CocoLabeledImage) Features() dlframework.Features {
 	return d.features
 }
 
+// Close ...
 func (d *CocoValidationTFRecord) Close() error {
 	if d.recordReader != nil {
 		d.recordReader.Close()
 	}
 	return nil
 }
 
+// Name ...
 func (d *CocoValidationTFRecord) Name() string {
 	return d.name
 }
 
+// CanonicalName ...
 func (d *CocoValidationTFRecord) CanonicalName() string {
 	category := strings.ToLower(d.Category())
 	name := strings.ToLower(d.Name())
@@ -92,6 +96,7 @@ func (d *CocoValidationTFRecord) workingDir() string {
 	return filepath.Join(d.baseWorkingDir, category, name)
 }
 
+// Download ...
 func (d *CocoValidationTFRecord) Download(ctx context.Context) error {
 	workingDir := d.workingDir()
 	fileName := d.recordFileName
@@ -110,14 +115,17 @@ func (d *CocoValidationTFRecord) Download(ctx context.Context) error {
 	return nil
 }
 
+// New ...
 func (d *CocoValidationTFRecord) New(ctx context.Context) (dldataset.Dataset, error) {
 	return nil, nil
 }
 
+// Get ...
 func (d *CocoValidationTFRecord) Get(ctx context.Context, name string) (dldataset.LabeledData, error) {
 	return nil, errors.New("get is not implemented for " + d.CanonicalName())
 }
 
+// List ...
 func (d *CocoValidationTFRecord) List(ctx context.Context) ([]string, error) {
 	return nil, errors.New("list is not implemented for " + d.CanonicalName())
 }
@@ -137,6 +145,7 @@ func (d *CocoValidationTFRecord) loadRecord(ctx context.Context) error {
 	return nil
 }
 
+// Load ...
 func (d *CocoValidationTFRecord) Load(ctx context.Context) error {
 	return d.loadRecord(ctx)
 }