From 944d78c6201c9845a20aaf8a93332fafa529b45b Mon Sep 17 00:00:00 2001 From: Brad Deam Date: Wed, 21 Jun 2023 12:10:47 +0930 Subject: [PATCH] Fix 'data-stream-stats' telemetry device version checks --- esrally/telemetry.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/esrally/telemetry.py b/esrally/telemetry.py index 403af4c84..86ba574d4 100644 --- a/esrally/telemetry.py +++ b/esrally/telemetry.py @@ -1351,9 +1351,9 @@ def __init__(self, telemetry_params, clients, metrics_store): def on_benchmark_start(self): for cluster_name in self.specified_cluster_names: recorder = DataStreamStatsRecorder(cluster_name, self.clients[cluster_name], self.metrics_store, self.sample_interval) - client_info = self.clients[cluster_name].info() - distribution_version = client_info["version"]["number"] - distribution_flavor = client_info["version"].get("build_flavor", "oss") + es_info = self.clients[cluster_name].info() + distribution_version = es_info["version"].get("number", "7.9.0") + distribution_flavor = es_info["version"].get("build_flavor", "oss") if Version.from_string(distribution_version) < Version(major=7, minor=9, patch=0): raise exceptions.SystemSetupError( "The data-stream-stats telemetry device can only be used with clusters from version 7.9 onwards"