diff --git a/setup.py b/setup.py index 52399ed..96deb2b 100644 --- a/setup.py +++ b/setup.py @@ -8,4 +8,8 @@ author_email='pavel@surmenok.com', license='MIT', packages=['keras_lr_finder'], + install_requires=[ + 'keras>=2.0.0', + 'matplotlib' + ], zip_safe=False) \ No newline at end of file