Update BertForJointParsing.py

BertForJointParsing.py

@@ -458,6 +458,10 @@ def convert_output_to_ud(output_sentences, model_cfg, style: Literal['htb', 'iah
                     # for HTB, if the function is poss, then add a shel pointing to the next word
                     if func == 'nmod:poss' and s_lex != 'של':
                         intermediate_output.append(dict(word='_של_', lex='של', pos='ADP', dep=len(intermediate_output) + 2, func='case', feats='_', absolute_dep=True))
+                    # if the function is obj, then add a את pointing to the next word
+                    elif func == 'obj' and s_lex != 'את':
+                        intermediate_output.append(dict(word='_את_', lex='את', pos='ADP', dep=len(intermediate_output) + 2, func='case', feats='_', absolute_dep=True))
+
                 # add the main suffix in
                 intermediate_output.append(dict(word=suf, lex='הוא', pos='PRON', dep=dep, func=func, feats='|'.join(f'{k}={v}' for k,v in word['morph']['suffix_feats'].items())))
                 if entry_to_assign_suf_dep: