Fixed templating display issues

mgr_bias_scoring.py

@@ -277,8 +277,8 @@ def maskSentenceDifferences(sentence, rewrite, target_words, att_term):
       rewrite = rewrite.replace(no_space_att, att_term)
 
   # identify group term in both sentences
-  sentence = sentence_to_template(sentence, target_words[0], "[T]")
-  rewrite = sentence_to_template(rewrite, target_words[1], "[T]")
+  sentence = sentence_to_template(sentence, target_words[0], "*")
+  rewrite = sentence_to_template(rewrite, target_words[1], "*")
   #print(f'S1: {sentence}')
   #print(f'R1: {rewrite}')
 
@@ -301,7 +301,17 @@ def maskSentenceDifferences(sentence, rewrite, target_words, att_term):
       else:
         template += "[R]"
         template_tokens.append("[R]")
-        add_refs.append((s.lower(),r.lower()))
+        
+        l_mask = s.lower()
+        r_mask = r.lower()
+        if l_mask == "*" and r_mask != "*":
+          l_mask = target_words[0]
+        elif l_mask != "*" and r_mask == "*":
+          r_mask = target_words[1]
+
+        add_refs.append((l_mask, r_mask))
+
+        #add_refs.append((s.lower(),r.lower()))
     elif s in string.punctuation:
       template += s.strip(" ")
       template_tokens.append(s)
@@ -311,7 +321,7 @@ def maskSentenceDifferences(sentence, rewrite, target_words, att_term):
 
     template += " "
 
-  return TreebankWordDetokenizer().detokenize(template_tokens), add_refs
+  return TreebankWordDetokenizer().detokenize(template_tokens).replace("*","[T]"), add_refs
 
 # turn generated sentence into a test templates - reference term aware version
 def ref_terms_sentence_to_template(row):