From 94f03d09f3645dcd807474ed371471a34a43a043 Mon Sep 17 00:00:00 2001
From: Steve Chien <schien@google.com>
Date: Tue, 29 Jun 2021 10:57:29 -0700
Subject: [PATCH] Add attrs to setup and requirements.

PiperOrigin-RevId: 382121530
---
 requirements.txt | 5 +++--
 setup.py         | 1 +
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index d5d560b..955d819 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,6 @@
 tensorflow>=1.14
-tensorflow-estimator>=2.3.0
-mpmath
 scipy>=0.17
+tensorflow-estimator>=2.3.0
+attrs>=21.2.0
+mpmath
 dm-tree~=0.1.1
diff --git a/setup.py b/setup.py
index 1a847f6..32ef7db 100644
--- a/setup.py
+++ b/setup.py
@@ -23,6 +23,7 @@ setup(
     install_requires=[
         'scipy>=0.17',
         'tensorflow-estimator>=2.3.0',  # for DP versions of estimator.
+        'attrs>=21.2.0',  # for tree_aggregation_query.py.
         'mpmath',  # used in tests only
         'dm-tree~=0.1.1',  # used in tests only
     ],