From d2d9f6ae0f4a58984e81ffc90931f6acb82254d0 Mon Sep 17 00:00:00 2001
From: Grzegorz Banasiak <grzegorz.banasiak@elastic.co>
Date: Wed, 29 May 2024 15:05:59 +0200
Subject: [PATCH] Fix comments

---
 esrally/client/factory.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/esrally/client/factory.py b/esrally/client/factory.py
index f7f487d9b..2b5b99e3e 100644
--- a/esrally/client/factory.py
+++ b/esrally/client/factory.py
@@ -15,6 +15,7 @@
 # specific language governing permissions and limitations
 # under the License.
 
+import contextlib
 import logging
 import time
 
@@ -363,12 +364,13 @@ def cluster_distribution_version(hosts, client_options, client_factory=EsClientF
     # if version number is not available default to build flavor
     version_number = version.get("number", version_build_flavor)
 
+    # assume non-operator serverless privileges by default
     serverless_operator = False
     if versions.is_serverless(version_build_flavor):
         # overwrite static serverless version number
         version_number = "serverless"
 
-        # determine operator status if security enabled, otherwise assume operator privileges
+        # determine serverless operator status if security enabled
         # pylint: disable=import-outside-toplevel
         from elasticsearch.exceptions import ApiError