fix: hardcode category to be batch

oom-ai · Nov 12, 2021 · e725d31 · e725d31
1 parent 22ad68b
commit e725d31
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pkg/oomstore/feature_group.go b/pkg/oomstore/feature_group.go
@@ -4,10 +4,14 @@ import (
 	"context"
 
 	"github.com/oom-ai/oomstore/internal/database/metadatav2"
+	"github.com/oom-ai/oomstore/pkg/oomstore/types"
 	"github.com/oom-ai/oomstore/pkg/oomstore/typesv2"
 )
 
 func (s *OomStore) CreateFeatureGroup(ctx context.Context, opt metadatav2.CreateFeatureGroupOpt) (int16, error) {
+	// Via the oomstore API, we can only create a batch feature group
+	// So we hardcode the category to be batch
+	opt.Category = types.BatchFeatureCategory
 	return s.metadatav2.CreateFeatureGroup(ctx, opt)
 }