Add comment

feathr-ai · Oct 13, 2022 · 074b6bc · 074b6bc
1 parent 42d0a1c
commit 074b6bc
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/main/scala/com/linkedin/feathr/offline/transformation/FDSConversionUtils.scala b/src/main/scala/com/linkedin/feathr/offline/transformation/FDSConversionUtils.scala
@@ -253,6 +253,8 @@ private[offline] object FDSConversionUtils {
         values.asScala.toArray
       case values: mutable.WrappedArray[Any] =>
         if (values.nonEmpty && values(0).isInstanceOf[GenericRowWithSchema]) {
+          // Assuming the result is returned by SWA feature with groupBy, hence keeping only the
+          // feature value as an array and dropping the index info.
           values.asInstanceOf[mutable.WrappedArray[GenericRowWithSchema]].map(v => v.get(v.size - 1)).toArray
         } else {
           values.toArray