Minor fixes - 2023 happy new year

• Added assert to Attention class (from extensions) when mask is used
• Fixed confusion matrix cpu/gpu device error
• Better handling on callbacks where apply_on_states=None (apply on all states)
• Updated Pipfile