incorporated review comments

Signed-off-by: msvinaykumar <[email protected]>
kruize · Dec 9, 2024 · ce6b674 · ce6b674
1 parent 0bad108
commit ce6b674
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/src/main/java/com/autotune/analyzer/recommendations/engine/RecommendationEngine.java b/src/main/java/com/autotune/analyzer/recommendations/engine/RecommendationEngine.java
@@ -197,9 +197,13 @@ private KruizeObject createKruizeObject(String target_cluster) {
         KruizeObject kruizeObject = new KruizeObject();
         try {
 
-            if (KruizeDeploymentInfo.is_ros_enabled && null != target_cluster && target_cluster.equalsIgnoreCase(AnalyzerConstants.REMOTE)) { // todo call this in function and use across every where
-                new ExperimentDBService().loadExperimentFromDBByName(mainKruizeExperimentMAP, experimentName);
-            } else {
+            if (KruizeDeploymentInfo.is_ros_enabled){
+                if(null == target_cluster ||  target_cluster.equalsIgnoreCase(AnalyzerConstants.REMOTE)){
+                    new ExperimentDBService().loadExperimentFromDBByName(mainKruizeExperimentMAP, experimentName);
+                }else{
+                    new ExperimentDBService().loadLMExperimentFromDBByName(mainKruizeExperimentMAP, experimentName);
+                }
+            }else{
                 new ExperimentDBService().loadLMExperimentFromDBByName(mainKruizeExperimentMAP, experimentName);
             }