fix library imports

frankaging · frankaging · commit e78348f25efa · 2024-01-12T04:09:05.000-08:00
diff --git a/requirements.txt b/requirements.txt
@@ -2,9 +2,8 @@ torch
 transformers
 datasets
 notebook
-protobuf==3.19.1
+protobuf==3.20.*
 matplotlib
-wandb
 seaborn
 ipywidgets
 plotnine
diff --git a/setup.py b/setup.py
@@ -10,7 +10,7 @@
 
 setup(
     name="pyvene",
-    version="0.0.3",
+    version="0.0.4",
     description="Use Activation Intervention to Interpret Causal Mechanism of Model",
     long_description=long_description,
     long_description_content_type='text/markdown',
