Update requirements: fix failing test

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.96k stars 167 forks source link

Update requirements: fix failing test #112

Open SumanthRH opened 7 months ago

SumanthRH commented 7 months ago

Hi @jalammar , looks like I missed a commit on #110. All updates are on 68d893150d4fe81fcb34ebb39bf154d7aa437d6d . Changes since last time:

change import of top_k_top_p_filtering

Use raw string in regex search to get rid of DeprecationWarning:

DeprecationWarning: invalid escape sequence '\.'
layer_number = re.search("(?<=\.)\d+(?=\.)", name).group(0)

I've run pytest tests locally (python 3.10) and all tests pass. This should work now!

SumanthRH commented 7 months ago

Ping @jalammar. I've also added a small test for bfloat16 and it works!

Also, I believe the pre-commit hooks need to be updated (had some issues), so I will add that in another PR.