add type hint for values in calculate_rsd args

Signed-off-by: Michael Oviedo <[email protected]>
opensearch-project · Oct 24, 2024 · c0a25c5 · c0a25c5
1 parent 4912799
commit c0a25c5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/osbenchmark/aggregator.py b/osbenchmark/aggregator.py
@@ -214,7 +214,7 @@ def calculate_weighted_average(self, task_metrics: Dict[str, List[Any]], iterati
 
         return weighted_metrics
 
-    def calculate_rsd(self, values, metric_name: str):
+    def calculate_rsd(self, values: List[Union[int, float]], metric_name: str):
         if not values:
             raise ValueError(f"Cannot calculate RSD for metric '{metric_name}': empty list of values")
         if len(values) == 1: