Steps to Reproduce

opts.at_edge is not called when float_win_behavior = 'mux'

Minimal Reproduction

{
	float_win_behavior = 'mux',
	at_edge = function(context)
     	print('foo')
	end,
}

1. Use the previous config.
2. Open a floating window.
3. Invoke the keymap that corresponds to move_cursor_right().
4. .
   EXPECTED: foo should be printed
   ACTUAL: foo is not printed

Cause

Because we are returning early:

We never reach the point where at_edge() is called:

Versions

nvim v0.10.0
kitty 0.35.1