From 7e3161c024816bba56fdddc8de50b58c51ed036f Mon Sep 17 00:00:00 2001 From: Lucas Roberts Date: Mon, 11 Nov 2024 14:43:26 -0500 Subject: [PATCH] Update setup.py to include IPython dep The package expects the IPython library to be available but does not include in the dependency list under `setup.py`'s 'install_requires=[...]` list of string literals. This commit adds the dependency. Prior to this the work around needed is to-in a shell-in the python env run `pip install IPython`-after running `pip intall bertviz` and then do your stuffs. --- setup.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 79487b7..ceb546a 100644 --- a/setup.py +++ b/setup.py @@ -19,5 +19,5 @@ license="Apache 2.0", packages=["bertviz"], include_package_data=True, - install_requires=["transformers>=2.0", "torch>=1.0", "tqdm", "boto3", "requests", "regex", "sentencepiece"], -) \ No newline at end of file + install_requires=["transformers>=2.0", "torch>=1.0", "tqdm", "boto3", "requests", "regex", "sentencepiece", "IPython"], +)