From 6f0f7481ab66720e9782de96795bd3003400731e Mon Sep 17 00:00:00 2001
From: Zach Carmichael <craymichael@meta.com>
Date: Thu, 20 Feb 2025 18:18:13 -0800
Subject: [PATCH] Add necessary deps to environment.yml (#1510)

Summary:
Pull Request resolved: https://github.com/pytorch/captum/pull/1510

Dependencies updated for conda env file in line with setup.py and meta.yaml.

Reviewed By: aobo-y

Differential Revision: D69954075

fbshipit-source-id: 82d6e16576d617eef1bc860ff9f6007bfaac9b67
---
 environment.yml | 3 +++
 1 file changed, 3 insertions(+)

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