Added an initial solution

utbot-analytics/src/main/kotlin/org/utbot/AnalyticsSetUp.kt

@@ -0,0 +1,12 @@
+package org.utbot
+
+import org.utbot.analytics.EngineAnalyticsContext
+import org.utbot.predictors.StateRewardPredictorFactoryImpl
+
+class AnalyticsSetUp {
+    companion object {
+        init {
+            EngineAnalyticsContext.stateRewardPredictorFactory[1] = StateRewardPredictorFactoryImpl()
+        }
+    }
+}

utbot-framework/src/main/kotlin/org/utbot/analytics/EngineAnalyticsContext.kt

@@ -6,6 +6,7 @@ import org.utbot.engine.selectors.NNRewardGuidedSelectorWithRecalculationFactory
 import org.utbot.engine.selectors.NNRewardGuidedSelectorWithoutRecalculationFactory
 import org.utbot.framework.NNRewardGuidedSelectorType
 import org.utbot.framework.UtSettings
+import java.util.PriorityQueue
 
 /**
  * Class that stores all objects that need for work analytics module during symbolic execution
@@ -28,9 +29,13 @@ object EngineAnalyticsContext {
         NNRewardGuidedSelectorType.WITH_RECALCULATION -> NNRewardGuidedSelectorWithRecalculationFactory()
     }
 
-    var stateRewardPredictorFactory: StateRewardPredictorFactory = object : StateRewardPredictorFactory {
-        override fun invoke(): StateRewardPredictor {
-            error("NNStateRewardPredictor factory wasn't provided")
+    var stateRewardPredictorFactory: MutableMap<Int, StateRewardPredictorFactory> = mutableMapOf()
+
+    init {
+        stateRewardPredictorFactory[0] = object : StateRewardPredictorFactory {
+            override fun invoke(): StateRewardPredictor {
+                error("NNStateRewardPredictor factory wasn't provided")
+            }
         }
     }
 }

utbot-intellij/build.gradle

@@ -27,7 +27,7 @@ dependencies {
 
     implementation(project(":utbot-framework")) { exclude group: 'org.slf4j', module: 'slf4j-api' }
     implementation(project(":utbot-fuzzers"))
-    api project(':utbot-analytics') // TODO???
+    implementation project(':utbot-analytics')
     testImplementation 'org.mock-server:mockserver-netty:5.4.1'
     testImplementation(project(":utbot-sample"))
     testApi(project(":utbot-framework"))

utbot-intellij/src/main/kotlin/org/utbot/intellij/plugin/generator/UtTestsDialogProcessor.kt

@@ -25,6 +25,7 @@ import com.intellij.testIntegration.TestIntegrationUtils
 import com.intellij.util.concurrency.AppExecutorUtil
 import mu.KotlinLogging
 import org.jetbrains.kotlin.idea.util.module
+import org.utbot.AnalyticsSetUp
 import org.utbot.analytics.EngineAnalyticsContext
 import org.utbot.analytics.Predictors
 import org.utbot.engine.util.mockListeners.ForceMockListener
@@ -51,16 +52,13 @@ import java.nio.file.Paths
 import java.util.concurrent.TimeUnit
 import org.utbot.common.filterWhen
 import org.utbot.engine.util.mockListeners.ForceStaticMockListener
-import org.utbot.features.FeatureExtractorFactoryImpl
-import org.utbot.features.FeatureProcessorWithStatesRepetitionFactory
 import org.utbot.framework.PathSelectorType
 import org.utbot.framework.plugin.api.testFlow
 import org.utbot.intellij.plugin.settings.Settings
 import org.utbot.intellij.plugin.ui.utils.isGradle
 import org.utbot.intellij.plugin.ui.utils.suitableTestSourceRoots
 import org.utbot.intellij.plugin.util.isAbstract
 import org.utbot.intellij.plugin.ui.utils.testModules
-import org.utbot.predictors.StateRewardPredictorFactoryImpl
 import kotlin.reflect.KClass
 import kotlin.reflect.full.functions
 
@@ -263,11 +261,12 @@ object UtTestsDialogProcessor {
     }
 
     private fun configureML() {
-        EngineAnalyticsContext.featureProcessorFactory = FeatureProcessorWithStatesRepetitionFactory()
-        EngineAnalyticsContext.featureExtractorFactory = FeatureExtractorFactoryImpl()
-        EngineAnalyticsContext.stateRewardPredictorFactory = StateRewardPredictorFactoryImpl()
+       // EngineAnalyticsContext.featureProcessorFactory = FeatureProcessorWithStatesRepetitionFactory()
+       // EngineAnalyticsContext.featureExtractorFactory = FeatureExtractorFactoryImpl()
+       // EngineAnalyticsContext.stateRewardPredictorFactory = StateRewardPredictorFactoryImpl()
+        //AnalyticsSetUp
         if (UtSettings.pathSelectorType == PathSelectorType.NN_REWARD_GUIDED_SELECTOR) {
-            Predictors.stateRewardPredictor = EngineAnalyticsContext.stateRewardPredictorFactory()
+            Predictors.stateRewardPredictor = EngineAnalyticsContext.stateRewardPredictorFactory[1]!!.invoke()
         }
 
         logger.info { "PathSelectorType: ${UtSettings.pathSelectorType}" }