Update sub-packages/bionemo-llm/src/bionemo/llm/utils/callbacks.py

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com>
Signed-off-by: John St. John <jstjohn@users.noreply.github.com>

Author: jstjohn

sub-packages/bionemo-llm/src/bionemo/llm/utils/callbacks.py

@@ -16,7 +16,11 @@
 
 import logging
 import os
-from typing import Any, Literal, Sequence, override
+from typing import Any, Literal, Sequence
+try:  # Python 3.12+
+    from typing import override
+except ImportError:  # Python < 3.12
+    from typing_extensions import override
 
 import lightning.pytorch as pl
 import torch