Update GANDLF/utils/generic.py

Co-authored-by: Sarthak Pati <patis@iu.edu>

GANDLF/utils/generic.py

@@ -287,26 +287,28 @@ def __update_metric_from_list_to_single_string(input_metrics_dict) -> dict:
 
 def define_average_type_key(
     params: Dict[str, Union[Dict[str, Any], Any]], metric_name: str
-):
+) -> str:
     """Determine if the the 'average' filed is defined in the metric config.
     If not, fallback to the default 'macro'
     values.
     Args:
         params (dict): The parameter dictionary containing training and data information.
         metric_name (str): The name of the metric.
+
     Returns:
         str: The average type key.
     """
     average_type_key = params["metrics"][metric_name].get("average", "macro")
     return average_type_key
 
 
-def define_multidim_average_type_key(params, metric_name):
+def define_multidim_average_type_key(params, metric_name) -> str:
     """Determine if the the 'multidim_average' filed is defined in the metric config.
     If not, fallback to the default 'global'.
     Args:
         params (dict): The parameter dictionary containing training and data information.
         metric_name (str): The name of the metric.
+
     Returns:
         str: The average type key.
     """