Updated setup.py to add easydict and absl requirements
@@ -10,14 +10,16 @@ setup(name='cleverhans',
'pycodestyle',
'scipy',
'matplotlib',
- "mnist",
- "numpy",
- "tensorflow-probability",
- "joblib",
+ 'mnist',
+ 'numpy',
+ 'tensorflow-probability',
+ 'joblib',
+ 'easydict',
+ 'absl'
],
extras_require={
- "jax": ["jax>=0.2.9", "jaxlib"],
- "tf": ["tensorflow>=2.4.0", "tensorflow-probability"],
- "pytorch": ["torch>=1.7.0", "torchvision>=0.8.0"],
+ 'jax': ['jax>=0.2.9', 'jaxlib'],
+ 'tf': ['tensorflow>=2.4.0', 'tensorflow-probability'],
+ 'pytorch': ['torch>=1.7.0', 'torchvision>=0.8.0'],
},
packages=find_packages())
