Shouldn't this be `np.sqrt(sig)` ? https://github.com/banyikun/EE-Net-ICLR-2022/blob/08d6ed7bca8f3818dbf4c502887bec2912f614a8/baselines/LinUCB.py#L18