diff --git a/tests/benchmark/milvus_benchmark/metrics/models/metric.py b/tests/benchmark/milvus_benchmark/metrics/models/metric.py
index 34d52040b8..7903e63cce 100644
--- a/tests/benchmark/milvus_benchmark/metrics/models/metric.py
+++ b/tests/benchmark/milvus_benchmark/metrics/models/metric.py
@@ -34,6 +34,7 @@ class Metric(object):
         self.run_id = int(time.time())
 
     def set_mode(self, mode):
+        # Set the deployment mode of milvus
         self.mode = mode
 
     # including: metric, suite_metric