Add attrs to setup and requirements.

PiperOrigin-RevId: 382121530
This commit is contained in:
Steve Chien 2021-06-29 10:57:29 -07:00 committed by A. Unique TensorFlower
parent 392c506c62
commit 94f03d09f3
2 changed files with 4 additions and 2 deletions

View file

@ -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

View file

@ -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
],