Keep device awake when using keyboard

At the moment the device screen will go to sleep while using the
keyboard while on the lock screen. This CL will align the
behavior of touch and key pressing.

Bug: 325932166
Test: Flashed on device
Flag: NA
Change-Id: I9293eca29dd91b2bf51d997ebb915bba14e3402f