In a MudBlazor DataGrid, how can I use inline editing and also use TemplateColumn?

I'm using a MudBlazor DataGrid with cells that use TemplateColumn for custom color. I would like to use EditMode="DataGridEditMode.Cell" for inline editing, but when I do I lose the TemplateColumn.

Is there a way to turn on inline editing only for the selected row? Using DataGridEditTrigger.OnRowClick" does not work.

https://try.mudblazor.com/snippet/wYcIPPcvKUrbAtTK