Hello,
I have been having the same issue with my Note 4.
After troubleshooting, I found out that Peel Smart Remote was causing it.
It came with my phone, but it updated itself, and the update added a little "peel remote" shortcut onto my locked screen, and it caused the screen to stay on after...