diff --git a/metaflow/plugins/aip/aip_cli.py b/metaflow/plugins/aip/aip_cli.py index d29bd2ba8d3..6142b49c7e6 100644 --- a/metaflow/plugins/aip/aip_cli.py +++ b/metaflow/plugins/aip/aip_cli.py @@ -414,14 +414,15 @@ def run( def _get_flow_parameters(kwargs, obj) -> Dict[str, Any]: + # AIP-9061(talebz): Black lowercases kwargs parameter names, so we call lower() def _convert_value(param: parameters.Parameter): - v = kwargs.get(param.name) + v = kwargs.get(param.name.lower()) return json.dumps(v) if param.kwargs.get("type") == JSONType else v flow_parameters = { param.name: _convert_value(param) for _, param in obj.flow._get_parameters() - if kwargs.get(param.name) is not None + if kwargs.get(param.name.lower()) is not None } return flow_parameters