Add matplotlib test dependency so we can run notebook_utils_test.

PiperOrigin-RevId: 454823630 Change-Id: I9903ca5eb2284c8775631e2ffaf452869c7d4d7c

Add matplotlib test dependency so we can run notebook_utils_test.
PiperOrigin-RevId: 454823630 Change-Id: I9903ca5eb2284c8775631e2ffaf452869c7d4d7c
725e79ab · Tom Ward · Copybara-Service · 5f4a9cc3 · 725e79ab
Commit 725e79ab authored Jun 14, 2022 by Tom Ward Committed by Copybara-Service Jun 14, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

setup.py setup.py +4 -1

No files found.
--- a/setup.py
+++ b/setup.py
@@ -42,7 +42,10 @@ setup(
        'scipy',
        'tensorflow-cpu',
    ],
-    tests_require=['mock'],
+    tests_require=[
+        'matplotlib',  # For notebook_utils_test.
+        'mock',
+    ],
    classifiers=[
        'Development Status :: 5 - Production/Stable',
        'Intended Audience :: Science/Research',