From c2925c9da267d49c62067ccecbe6cb6f8606033d Mon Sep 17 00:00:00 2001
From: Jiyao Li <jiyao.thomas.li@gmail.com>
Date: Wed, 24 Nov 2021 16:45:23 -0700
Subject: [PATCH] add h5py in requirement.txt and add conda intall atari in
 readme

---
 README.md        | 3 +++
 requirements.txt | 3 ++-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 5784361b1..365555890 100644
--- a/README.md
+++ b/README.md
@@ -53,6 +53,9 @@ conda install pytorch torchvision -c soumith
 
 # Other requirements
 pip install -r requirements.txt
+
+# Gym Atari
+conda install -c conda-forge gym-atari
 ```
 
 ## Contributions
diff --git a/requirements.txt b/requirements.txt
index b996ad420..ed4d617e2 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,5 @@
 gym
 matplotlib
 pybullet
-stable-baselines3
\ No newline at end of file
+stable-baselines3
+h5py
\ No newline at end of file