I was able to reproduce the issue and it appears to be caused by how OnePlus implements their battery management, which is non-compliant with the way Google implements wake locks in Android. You can read more here: https://dontkillmyapp.com/oneplus
From my tests the only way to prevent this issue is by having the phone connected to a battery pack.