Correct shape in test_activations_dict_to_array input example

jalammar / ecco

Explain, analyze, and visualize NLP language models. Ecco creates interactive visualizations directly in Jupyter notebooks explaining the behavior of Transformer-based language models (like GPT2, BERT, RoBERTA, T5, and T0).

https://ecco.readthedocs.io

BSD 3-Clause "New" or "Revised" License

1.98k stars 168 forks source link

Correct shape in test_activations_dict_to_array input example #13

Closed nostalgebraist closed 3 years ago

nostalgebraist commented 3 years ago

Fix shape of test input in test_activations_dict_to_array.

See this comment for details.

Confirmed that all tests pass locally, with:

cd ecco
pip install -r requirements.txt
pip install -e .

python -m pytest