From 238f3125dfdbad20c72f382ce9e915acbe9e13ab Mon Sep 17 00:00:00 2001
From: Maja Jablonska <majajjablonska@gmail.com>
Date: Sat, 11 Nov 2023 21:27:29 +1100
Subject: [PATCH] Fix an import in combo_model.py
---
combo/combo_model.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/combo/combo_model.py b/combo/combo_model.py
index 5372992..23eff70 100644
--- a/combo/combo_model.py
+++ b/combo/combo_model.py
@@ -22,6 +22,7 @@ from combo.data import Instance
from combo.data.batch import Batch
from combo.data.dataset_loaders.dataset_loader import TensorDict
from combo.nn import utils
+from combo.predictors import Predictor
@Registry.register("semantic_multitask")
@@ -42,7 +43,7 @@ class ComboModel(Model, FromParameters):
lemmatizer: LemmatizerModel = None,
upos_tagger: MorphologicalFeatures = None,
xpos_tagger: MorphologicalFeatures = None,
- semantic_relation: base.Predictor = None,
+ semantic_relation: Predictor = None,
morphological_feat: MorphologicalFeatures = None,
dependency_relation: DependencyRelationModel = None,
enhanced_dependency_relation: DependencyRelationModel = None,
--
GitLab