Address PR comment

#546 (comment)
elastic · Aug 7, 2018 · 581a69f · 581a69f
1 parent 25ae686
commit 581a69f
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/esrally/mechanic/telemetry.py b/esrally/mechanic/telemetry.py
@@ -411,12 +411,11 @@ def record_stats_per_index(self, name, stats):
         """
 
         for shard_stats in stats:
-            shard_id = shard_stats.get("shard_id", None)
-            if shard_id is not None:
+            if "shard_id" in shard_stats:
                 shard_metadata = {
                     "cluster": self.cluster_name,
                     "index": name,
-                    "shard": shard_id,
+                    "shard": shard_stats["shard_id"],
                     "name": "ccr-stats"
                 }