Use androidx Flaky annotation for device tests.

Platform_testing flaky annotation is for
host side tests and will be moved, device
tests should use androidx annotation.

(and fix a couple ktlint spacing issues)

Bug: 293837621
Test: presubmit
Change-Id: I91c58dcfbec623a98647ded67dff2e9d7d90d5a4