From 6aa31d95ddd94588ccd151544610d622731593e7 Mon Sep 17 00:00:00 2001
From: Michael Reneer <michaelreneer@google.com>
Date: Thu, 10 Feb 2022 21:06:32 -0800
Subject: [PATCH] Add direct `matplotlib ` dependency.

PiperOrigin-RevId: 427914629
---
 requirements.txt | 1 +
 setup.py         | 1 +
 2 files changed, 2 insertions(+)

diff --git a/requirements.txt b/requirements.txt
index 04087d9..6dc0ba1 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,7 @@
 absl-py~=1.0.0
 attrs>=21.2.0
 dm-tree~=0.1.1
+matplotlib~=3.3.4
 mpmath
 scipy>=0.17
 tensorflow-datasets>=4.4.0
diff --git a/setup.py b/setup.py
index 6352890..9488908 100644
--- a/setup.py
+++ b/setup.py
@@ -30,6 +30,7 @@ setup(
         'absl-py~=1.0.0',
         'attrs>=21.2.0',
         'dm-tree~=0.1.1',
+        'matplotlib~=3.3.4',
         'scipy>=0.17',
         'tensorflow-datasets>=4.4.0',
         'tensorflow-estimator>=2.3.0',