Skip to content

Commit 5e1592d

Browse files
committed
assert MLIndicesHandler has non-null MLFeatureEnabledSettingObject
Signed-off-by: Brian Flores <[email protected]>
1 parent 53df199 commit 5e1592d

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

ml-algorithms/src/main/java/org/opensearch/ml/engine/indices/MLIndicesHandler.java

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
import java.util.Map;
1717
import java.util.concurrent.atomic.AtomicBoolean;
1818

19+
import lombok.NonNull;
1920
import org.opensearch.OpenSearchWrapperException;
2021
import org.opensearch.ResourceAlreadyExistsException;
2122
import org.opensearch.action.admin.indices.create.CreateIndexRequest;
@@ -44,9 +45,11 @@
4445
@RequiredArgsConstructor
4546
@Log4j2
4647
public class MLIndicesHandler {
47-
48+
@NonNull
4849
ClusterService clusterService;
50+
@NonNull
4951
Client client;
52+
@NonNull
5053
MLFeatureEnabledSetting mlFeatureEnabledSetting;
5154
private static final Map<String, AtomicBoolean> indexMappingUpdated = new HashMap<>();
5255

plugin/src/main/java/org/opensearch/ml/plugin/MachineLearningPlugin.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -599,6 +599,7 @@ public Collection<Object> createComponents(
599599
Settings settings = environment.settings();
600600
Path dataPath = environment.dataFiles()[0];
601601

602+
mlFeatureEnabledSetting = new MLFeatureEnabledSetting(clusterService, settings);
602603
mlIndicesHandler = new MLIndicesHandler(clusterService, client, mlFeatureEnabledSetting);
603604

604605
SdkClient sdkClient = SdkClientFactory
@@ -665,7 +666,6 @@ public Collection<Object> createComponents(
665666
mlInputDatasetHandler = new MLInputDatasetHandler(client);
666667
modelAccessControlHelper = new ModelAccessControlHelper(clusterService, settings);
667668
connectorAccessControlHelper = new ConnectorAccessControlHelper(clusterService, settings);
668-
mlFeatureEnabledSetting = new MLFeatureEnabledSetting(clusterService, settings);
669669
mlModelManager = new MLModelManager(
670670
clusterService,
671671
scriptService,

0 commit comments

Comments
 (0)