Update transformer_re_text_classification2.py

transformer_re_text_classification2.py

@@ -293,6 +293,10 @@ class TransformerRETextClassificationTaskModule2(_TransformerReTextClassificatio
         entities: Sequence[Span] = document[self.entity_annotation]
 
         relations: Sequence[BinaryRelation] = document[self.relation_annotation]
+        # if no relations are predefined, use None so that enumerate_entities yields all pairs
+        # should be fixed to a parameter "generate_all" or "restrict_to_existing_relations"
+        if len(relations) == 0:
+            relations = None
 
         partitions: Sequence[Optional[Span]]
         if self.partition_annotation is not None:
@@ -424,7 +428,8 @@ class TransformerRETextClassificationTaskModule2(_TransformerReTextClassificatio
 
                 # lots of logging from here on
                 log_this_example = (
-                    self.log_first_n_examples is not None
+                    relations is not None
+                    and self.log_first_n_examples is not None
                     and self._logged_examples_counter <= self.log_first_n_examples
                 )
                 if log_this_example: