Finishing off

droberts195 · Mar 20, 2024 · 190df62 · 190df62
1 parent 98d45f8
commit 190df62
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 10 deletions.
diff --git a/...src/main/java/org/elasticsearch/xpack/ml/action/TransportGetTrainedModelsStatsAction.java b/...src/main/java/org/elasticsearch/xpack/ml/action/TransportGetTrainedModelsStatsAction.java
@@ -214,10 +214,7 @@ protected void doExecuteForked(
                     responseBuilder.getExpandedModelIdsWithAliases(),
                     request.isAllowNoResources(),
                     parentTaskId,
-                    l.map(modelSizeStatsByModelId -> {
-                        responseBuilder.setModelSizeStatsByModelId(modelSizeStatsByModelId);
-                        return null;
-                    }),
+                    l,
                     numberOfAllocations
                 )
             )

diff --git a/...in/ml/src/main/java/org/elasticsearch/xpack/ml/utils/InferenceProcessorInfoExtractor.java b/...in/ml/src/main/java/org/elasticsearch/xpack/ml/utils/InferenceProcessorInfoExtractor.java
@@ -13,7 +13,6 @@
 import org.elasticsearch.ingest.ConfigurationUtils;
 import org.elasticsearch.ingest.IngestMetadata;
 import org.elasticsearch.ingest.Pipeline;
-import org.elasticsearch.logging.LogManager;
 import org.elasticsearch.transport.Transports;
 
 import java.util.HashMap;
@@ -77,11 +76,7 @@ public static int countInferenceProcessors(ClusterState state) {
      */
     @SuppressWarnings("unchecked")
     public static Map<String, Set<String>> pipelineIdsByResource(ClusterState state, Set<String> ids) {
-        try {
-            assert Transports.assertNotTransportThread("non-trivial nested loops over cluster state structures");
-        } catch (AssertionError e) {
-            LogManager.getLogger(InferenceProcessorInfoExtractor.class).fatal("should not be here", e);
-        }
+        assert Transports.assertNotTransportThread("non-trivial nested loops over cluster state structures");
         Map<String, Set<String>> pipelineIdsByModelIds = new HashMap<>();
         Metadata metadata = state.metadata();
         if (metadata == null) {