fixing output metadata

project-flogo · Mar 14, 2019 · f534fb0 · f534fb0
1 parent 0333920
commit f534fb0
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/activity/inference/activity.go b/activity/inference/activity.go
@@ -29,7 +29,7 @@ const (
 	ovResult = "result"
 )
 
-var activityMd = activity.ToMetadata(&Input{})
+var activityMd = activity.ToMetadata(&Input{}, &Output{})
 
 func init() {
 	activity.Register(&Activity{}, New)
@@ -65,7 +65,7 @@ func (a *Activity) Eval(context activity.Context) (done bool, err error) {
 	fw := context.GetInput(ivFramework).(string)
 
 	tfFramework := framework.Get(fw)
-	fmt.Println(tfFramework.FrameworkTyp())
+	log.Debugf("Using Framework %s", tfFramework.FrameworkTyp())
 	if tfFramework == nil {
 		log.Errorf("%s framework not registered", fw)
 
@@ -133,6 +133,7 @@ func (a *Activity) Eval(context activity.Context) (done bool, err error) {
 		for i := 0; i < len(classes); i++ {
 			out[classes[i]] = scores[i]
 		}
+
 		context.SetOutput(ovResult, out)
 	} else {
 		context.SetOutput(ovResult, output)