diff --git a/.conda/meta.yaml b/.conda/meta.yaml index f0bfc38b5b..c82b04eab6 100644 --- a/.conda/meta.yaml +++ b/.conda/meta.yaml @@ -28,8 +28,14 @@ test: about: home: https://captum.ai - license: BSD + license: BSD-3 license_file: LICENSE summary: Model interpretability for PyTorch + description: | + Captum is a model interpretability and understanding library for PyTorch. + Captum means comprehension in Latin and contains general purpose implementations + of integrated gradients, saliency maps, smoothgrad, vargrad and others for + PyTorch models. It has quick integration for models built with domain-specific + libraries such as torchvision, torchtext, and others. doc_url: https://captum.ai dev_url: https://github.com/pytorch/captum diff --git a/environment.yml b/environment.yml index e2a5275b45..fc7d864223 100644 --- a/environment.yml +++ b/environment.yml @@ -4,3 +4,6 @@ channels: dependencies: - numpy<2.0 - pytorch>=1.10 + - matplotlib-base + - tqdm + - packaging diff --git a/setup.py b/setup.py index 588f8f3a28..19003466b5 100755 --- a/setup.py +++ b/setup.py @@ -142,7 +142,7 @@ def get_package_files(root, subdirs): "PyTorch", ], classifiers=[ - "Development Status :: 4 - Beta", + "Development Status :: 5 - Production/Stable", "Intended Audience :: Developers", "Intended Audience :: Education", "Intended Audience :: Science/Research",