From 51b26df308ee5f13482df069cd7bb516faa09603 Mon Sep 17 00:00:00 2001
From: Juhan Bae <pomonam15@gmail.com>
Date: Tue, 9 Apr 2024 13:45:35 -0400
Subject: [PATCH] Fix setup dependency

---
 setup.cfg | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/setup.cfg b/setup.cfg
index 6fb1432..2bac67a 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -1,6 +1,6 @@
 [metadata]
 name = jax_influence
-version = 0.1.0
+version = 0.2.0
 description = If influence Functions are the Answer, Then What is the Question?
 url = https://github.com/pomonam/jax-influence
 author = Juhan Bae
@@ -32,15 +32,12 @@ install_requires =
   absl-py==1.0.0
   torch==1.12.1
   torchvision==0.13.1
-  numpy==1.21.6
-  scikit-learn==1.0.1
-  protobuf==3.20.*
-  clu==0.0.11
+  numpy==1.26.4
+  scikit_learn==1.4.0
   isort==5.10.1
   pylint==2.14.5
   pytest==7.1.2
   yapf==0.32.0
-  wandb==0.13.5
 python_requires = >=3.8
 
 [options.extras_require]