[Bug Report] 1 failing unit test on fresh install

[JasonBenn]

If you are submitting a bug report, please fill in the following details and use the tag [bug].

Describe the bug
poetry run pytest tests/unit/test_next_sentence_prediction.py::test_device_handling_to_tokens fails with UnboundLocalError: local variable 'token_type_ids' referenced before assignment

Code example
poetry run pytest tests/unit/test_next_sentence_prediction.py::test_device_handling_to_tokens

System Info
Describe the characteristic of your environment:

• Describe how transformer_lens was installed (pip, docker, source, ...): poetry, per the install instructions (I did not use a Dev Container)
• What OS are you using? (Linux, MacOS, Windows): Linux
• Python version (We support 3.7--3.10 currently): 3.10.12

Checklist

• I have checked that there is no similar issue in the repo (required)

[JasonBenn]

Fixed in #914