Skip to content

Commit b0473cb

Browse files
committed
Merge branch 'jstjohn/cp_support_predict' of github.com:NVIDIA/bionemo-framework into jstjohn/cp_support_predict
2 parents 9a7a5ef + 8b8592e commit b0473cb

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

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

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,11 @@
1515

1616

1717
import os
18-
from typing import Any, Literal, Sequence, override
18+
from typing import Any, Literal, Sequence
19+
try: # Python 3.12+
20+
from typing import override
21+
except ImportError: # Python < 3.12
22+
from typing_extensions import override
1923

2024
import lightning.pytorch as pl
2125
import torch

0 commit comments

Comments
 (0)