make gym-copter optional as used only for rendering

Author: acxz

gym_ccc/envs/multirotor.py

@@ -1,8 +1,13 @@
 """Continuous Multirotor Environment."""
+import sys
+
 import gym
 from gym import spaces
 
-import gym_copter.rendering.twod as twod
+try:
+    import gym_copter.rendering.twod
+except ImportError:
+    pass
 
 import numpy as np
 
@@ -341,7 +346,10 @@ def __init__(self, gravity=9.8, mass=1.0, dt=0.02, custom_reset=None):
                                             obs_high, dtype=np.float32)
 
         # Support for rendering
-        self.renderer = None
+        if 'gym_copter.rendering.twod' in sys.modules:
+            self.renderer = None
+        else:
+            self.renderer = False
         self.pose = None
 
     def step(self, action):
@@ -389,9 +397,14 @@ def reset(self):
 
     def render(self, mode='human'):
         """Show the current state."""
+        # Print out state instead of cool gui if renderer not available
+        if self.renderer is False:
+            print(self.state)
+            return self.state
+
         # Creater renderer
         if self.renderer is None:
-            self.renderer = twod.TwoDRenderer()
+            self.renderer = gym_copter.rendering.twod.TwoDRenderer()
 
         # Just a flag to show the props spinning
         flight_status = True

setup.py

@@ -1,18 +1,24 @@
 """Setup file."""
 
-from setuptools import setup
+import setuptools
 
-setup(
+with open('README.md', 'r') as fh:
+    long_description = fh.read()
+
+setuptools.setup(
     name='gym-ccc',
     version='0.0.1',
     author='acxz',
-    long_description='',
+    long_description=long_description,
     description='Environments for continuous classical control problems with' +
     'dynamical state information',
-    packages=['gym_ccc'],
+    packages=setuptools.find_packages(),
     install_requires=[
         'gym',
-        'gym-copter',
         'numpy',
+        'pyquaternion',
     ],
+    extra_requires={
+        'copter-render': ['gym-copter'],
+    }
 )