presto-ml/src/main/java/io/prestosql/plugin/ml/LearnStateFactory.java/LearnStateFactory/getGroupedStateClass
class LearnStateFactory:    @Override
    public Class<? extends LearnState> getGroupedStateClass()
    {
        return GroupedLearnState.class;
    }