chore(lint): enforce PEP 604 optional annotations (#3535)

* chore(lint): enforce PEP 604 optional annotations

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

* Update uv lock

Signed-off-by: NeMo Bot <nemo-bot@nvidia.com>

* fix(annotations): defer optional UCCL types

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

---------

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>
Signed-off-by: NeMo Bot <nemo-bot@nvidia.com>
Co-authored-by: NeMo Bot <nemo-bot@nvidia.com>