add record checking for allennlp and spacy

forte_wrapper/allennlp/allennlp_processors.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 
 import logging
-from typing import List, Dict
+from typing import List, Dict, Set
 
 from allennlp.predictors import Predictor
 from forte.common import ProcessorConfigError
@@ -186,3 +186,14 @@ def _create_srl(input_pack: DataPack, tokens: List[Token],
                                         tokens[arg_span.end].end)
                 link = PredicateLink(input_pack, pred, arg)
                 link.arg_type = label
+
+    @classmethod
+    def expected_types_and_attributes(cls) -> Dict[str, Set[str]]:
+        r"""Method to add expected type for current processor input which
+        would be checked before running the processor if
+        :meth:`~forte.pipeline.Pipeline.enforce_consistency` was enabled for
+        the pipeline.
+        """
+        expectation_dict: Dict[str, Set[str]] = dict()
+        expectation_dict["ft.onto.base_ontology.Sentence"] = set()
+        return expectation_dict

forte_wrapper/spacy/spacy_processors.py

@@ -11,7 +11,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-from typing import Optional
+from typing import Optional, Dict, Set
 
 import spacy
 from spacy.language import Language
@@ -125,3 +125,13 @@ def _process(self, input_pack: DataPack):
 
         # Process sentence parses.
         self._process_parser(result.sents, input_pack)
+
+    def record(self, record_meta: Dict[str, Set[str]]):
+        r"""Method to add output type record of current processor
+        to :attr:`forte.data.data_pack.Meta.record`.
+
+        Args:
+            record_meta: the field in the datapack for type record that need to
+                fill in for consistency checking.
+        """
+        record_meta["ft.onto.base_ontology.Sentence"] = set()