remove leftovers from #371: is_training parameter in taskmodule imple…

…mentations (#376)
ArneBinder · Nov 13, 2023 · 1a481d6 · 1a481d6
1 parent 65c3d53
commit 1a481d6
Show file tree

Hide file tree

Showing 6 changed files with 0 additions and 6 deletions.
diff --git a/src/pytorch_ie/taskmodules/simple_transformer_text_classification.py b/src/pytorch_ie/taskmodules/simple_transformer_text_classification.py
@@ -112,7 +112,6 @@ def _post_prepare(self):
     def encode_input(
         self,
         document: DocumentType,
-        is_training: bool = False,
     ) -> TaskEncodingType:
         """
         Create one or multiple task encodings for the given document.

diff --git a/src/pytorch_ie/taskmodules/transformer_re_text_classification.py b/src/pytorch_ie/taskmodules/transformer_re_text_classification.py
@@ -328,7 +328,6 @@ def _create_relation_candidates(
     def encode_input(
         self,
         document: TextDocument,
-        is_training: bool = False,
     ) -> Optional[Union[TaskEncodingType, Sequence[TaskEncodingType],]]:
         relations: Sequence[BinaryRelation]
         if self.create_relation_candidates:

diff --git a/src/pytorch_ie/taskmodules/transformer_seq2seq.py b/src/pytorch_ie/taskmodules/transformer_seq2seq.py
@@ -98,7 +98,6 @@ def encode_text(self, text: str) -> InputEncodingType:
     def encode_input(
         self,
         document: TextDocument,
-        is_training: bool = False,
     ) -> Optional[Union[TaskEncodingType, Sequence[TaskEncodingType],]]:
         return TaskEncoding(
             document=document,

diff --git a/src/pytorch_ie/taskmodules/transformer_span_classification.py b/src/pytorch_ie/taskmodules/transformer_span_classification.py
@@ -143,7 +143,6 @@ def prepare(self, documents: Sequence[TextDocument]) -> None:
     def encode_input(
         self,
         document: TextDocument,
-        is_training: bool = False,
     ) -> Optional[Union[TaskEncodingType, Sequence[TaskEncodingType],]]:
         partitions: Sequence[Span]
         if self.single_sentence:

diff --git a/src/pytorch_ie/taskmodules/transformer_text_classification.py b/src/pytorch_ie/taskmodules/transformer_text_classification.py
@@ -147,7 +147,6 @@ def prepare(self, documents: Sequence[TextDocument]) -> None:
     def encode_input(
         self,
         document: TextDocument,
-        is_training: bool = False,
     ) -> Optional[Union[TaskEncodingType, Sequence[TaskEncodingType],]]:
         inputs = self.tokenizer(
             document.text,

diff --git a/src/pytorch_ie/taskmodules/transformer_token_classification.py b/src/pytorch_ie/taskmodules/transformer_token_classification.py
@@ -156,7 +156,6 @@ def encode_text(
     def encode_input(
         self,
         document: TextDocument,
-        is_training: bool = False,
     ) -> Optional[Union[TaskEncodingType, Sequence[TaskEncodingType],]]:
         partitions: Sequence[Optional[Span]]
         if self.partition_annotation is not None: