Do you mean adding `metric_kwargs` that takes a dictionary as input, or just renaming `kwargs`?