fixed passing relation_distribution and relation_label_distribution

combo/predict.py

@@ -109,7 +109,8 @@ class COMBO(PredictorModule):
              relation_distribution, relation_label_distribution) = self._predictions_as_tree(prediction, instance)
             sentence = conllu2sentence(
                 tree, sentence_embedding, embeddings,
-                relation_distribution, relation_label_distribution
+                relation_distribution=relation_distribution,
+                relation_label_distribution=relation_label_distribution
             )
             sentences.append(sentence)
         return sentences
@@ -125,7 +126,11 @@ class COMBO(PredictorModule):
         predictions = super().predict_instance(instance)
         (tree, sentence_embedding, embeddings,
          relation_distribution, relation_label_distribution) = self._predictions_as_tree(predictions, instance)
-        return conllu2sentence(tree, sentence_embedding, embeddings, relation_distribution, relation_label_distribution)
+        return conllu2sentence(
+            tree, sentence_embedding, embeddings,
+            relation_distribution=relation_distribution,
+            relation_label_distribution=relation_label_distribution
+        )
 
     @overrides
     def predict_json(self, inputs: JsonDict) -> data.Sentence: