update: setup and dependency installation

README.md

@@ -12,3 +12,23 @@ WiseRL provides unofficial and banchmarked PyTorch implementations for Offline P
 - BT Model + IQL/AWAC (BT-IQL/AWAC)
 - Contrastive Prefereing Learning (CPL)
 - Inverse Preference Learning + IQL/AWAC (IPL-IQL/AWAC)
+
+# Installation
++ clone this repo and install the dependencies
+  ```bash
+  git clone git@github.com:typoverflow/WiseRL
+  cd WiseRL && pip install -e .
+  ```
++ install environment or dataset dependencies
+  + for D4RL experiments:
+    ```bash
+    git clone https://github.com/Farama-Foundation/d4rl.git
+    cd d4rl
+    pip install -e .
+    ```
+  + for metaworld experiments:
+    ```bash
+    git clone git@github.com:Farama-Foundation/Metaworld
+    cd Metaworld && git checkout 04be337a
+    pip install -e .
+    ```

setup.py

@@ -16,9 +16,13 @@ def get_install_requires():
     return [
         "gym>=0.23.1,<=0.24.1",
         "tqdm",
+        "h5py",
+        "Cython==0.29.36"
+        "imageio",
         "numpy",
         "torch",
-        "pandas",
+        "tensorboard",
+        "wandb",
         "UtilsRL"
     ]
 
@@ -44,6 +48,7 @@ def get_extras_require():
         "tqdm",
         "numpy",
         "torch",
+        "torchvision"
         "pandas",
         "UtilsRL"
     ]

wiserl/__init__.py

@@ -1,2 +1,2 @@
 
-__version__ = "0.0.1"
+__version__ = "0.0.2"