[Bug]: at_edge has no effect for floating windows when float_win_behavior = 'mux'

[thenbe]

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:

smart-splits.nvim/lua/smart-splits/api.lua

Lines 359 to 363 in 9a6cd46

	if handle_floating_window(function()
	mux.move_pane(direction, true, at_edge)
	end) then
	return
	end

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

smart-splits.nvim/lua/smart-splits/api.lua

Line 393 in 9a6cd46

at_edge(ctx)

Versions

nvim v0.10.0
kitty 0.35.1