Skip to content
This repository was archived by the owner on Jul 28, 2025. It is now read-only.

Commit f1fb88a

Browse files
committed
CU-8698ek477: Fix AdamW import (trf->torch)
1 parent 481836b commit f1fb88a

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

medcat/utils/meta_cat/ml_utils.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,8 @@
1414
from sklearn.metrics import classification_report, precision_recall_fscore_support, confusion_matrix
1515
from sklearn.model_selection import train_test_split
1616
from sklearn.utils.class_weight import compute_class_weight
17-
from transformers import AdamW, get_linear_schedule_with_warmup # TODO: import from troch.optim instead
17+
from transformers import get_linear_schedule_with_warmup
18+
from torch.optim import AdamW
1819

1920

2021
import logging

0 commit comments

Comments
 (0)