make demonstration util as optional imports

ARISE-Initiative · Oct 26, 2018 · 5f1d45d · 5f1d45d
1 parent db4a44f
commit 5f1d45d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/requirements-extra.txt b/requirements-extra.txt
@@ -6,3 +6,6 @@ gym
 
 # Mac OS X only
 hidapi
+
+# required for demonstration utils
+h5py
diff --git a/robosuite/__init__.py b/robosuite/__init__.py
@@ -8,7 +8,6 @@
 
 from robosuite.environments.baxter_lift import BaxterLift
 from robosuite.environments.baxter_peg_in_hole import BaxterPegInHole
-from robosuite.wrappers import DataCollectionWrapper
 
 __version__ = "0.1.0"
 __logo__ = """