[#41] fix wrong import

aisingapore · Dec 10, 2021 · e290ba4 · e290ba4
1 parent 9b8417b
commit e290ba4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sgnlp/models/sentic_asgcn/modeling.py b/sgnlp/models/sentic_asgcn/modeling.py
@@ -2,8 +2,8 @@
 
 import torch
 import torch.nn as nn
+from transformers import PreTrainedModel
 from transformers.file_utils import ModelOutput
-from transformers.utils.dummy_pt_objects import PreTrainedModel
 
 
 @dataclass