Use a UI policy or Client script to hide the edit button when the state is "Assess" or "Authorize"

Client Script for reference:

if (g_form.getValue('state') == 'Assess' || g_form.getValue('state') == 'Authorize') {
// Hide Edit button logic here

g_form.setVisible('edit_button', false); // Hides the button
}