Spaces:

alistairmcleay
/

cambridge-masters-project

Runtime error

alistairmcleay commited on Jun 29, 2022

Commit

f29dd07

•

1 Parent(s): 2db6fc4

Fix bug in lexicalisation code

Files changed (1) hide show

scripts/UBAR_code/interaction/UBAR_interact.py CHANGED Viewed

@@ -128,7 +128,7 @@ class UbarSystemModel:  # may inherit convlab or not, just like andy's
                         for idx, belief_state_slot in enumerate(decoded_belief_states):
                             if token in slots_to_db_keys_map.keys():
                                 if slots_to_db_keys_map[token] == belief_state_slot:
-                                    token == decoded_belief_states[idx + 1]
                     # Otherwise just leave the slot as it is as we have failed to fill it

                         for idx, belief_state_slot in enumerate(decoded_belief_states):
                             if token in slots_to_db_keys_map.keys():
                                 if slots_to_db_keys_map[token] == belief_state_slot:
+                                    token = decoded_belief_states[idx + 1]
                     # Otherwise just leave the slot as it is as we have failed to fill it