addressing comments

Signed-off-by: Dhrubo Saha <[email protected]>
opensearch-project · Oct 3, 2023 · 78c458b · 78c458b
1 parent 0bdf991
commit 78c458b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/plugin/src/main/java/org/opensearch/ml/action/stats/MLStatsNodeResponse.java b/plugin/src/main/java/org/opensearch/ml/action/stats/MLStatsNodeResponse.java
@@ -33,7 +33,7 @@ public class MLStatsNodeResponse extends BaseNodeResponse implements ToXContentF
      */
     private Map<FunctionName, MLAlgoStats> algorithmStats;
     /**
-     * Model stats which includes stats level stats.
+     * Model stats which includes model level stats.
      *
      * Example: {model_id: { predict: { request_count: 1} }}
      */

diff --git a/plugin/src/main/java/org/opensearch/ml/stats/MLModelStats.java b/plugin/src/main/java/org/opensearch/ml/stats/MLModelStats.java
@@ -36,7 +36,9 @@ public MLModelStats(Map<ActionName, MLActionStats> modelStats) {
 
     @Override
     public void writeTo(StreamOutput out) throws IOException {
-        out.writeMap(modelStats, (stream, v) -> stream.writeEnum(v), (stream, stats) -> stats.writeTo(stream));
+        if (modelStats != null && modelStats.size() > 0) {
+            out.writeMap(modelStats, (stream, v) -> stream.writeEnum(v), (stream, stats) -> stats.writeTo(stream));
+        }
     }
 
     @Override