Ok, so after 24 hours or so, there has been a noticeable improvement, so I am assuming it takes the phone a while to accurately predict remaining battery life post One UI 6.1. I wasn't sure because there is no indication that it is learning, similar to when you first set up a new device.