Merge pull request #39 from rohanpsingh/autopep8-patches

Fixes by autopep8 action

Author: rohanpsingh

examples/sample.py

@@ -7,9 +7,9 @@
 # create the viewer object
 viewer = mujoco_viewer.MujocoViewer(model, data)
 
-viewer.add_line_to_fig(line_name="root-pos-x", fig_idx = 0)
-viewer.add_line_to_fig(line_name="root-pos-z", fig_idx = 0)
-viewer.add_line_to_fig(line_name="right_ankle_y", fig_idx = 1)
+viewer.add_line_to_fig(line_name="root-pos-x", fig_idx=0)
+viewer.add_line_to_fig(line_name="root-pos-z", fig_idx=0)
+viewer.add_line_to_fig(line_name="right_ankle_y", fig_idx=1)
 
 # user has access to mjvFigure
 fig = viewer.figs[0]
@@ -29,8 +29,10 @@
 
 # simulate and render
 for _ in range(100000):
-    viewer.add_data_to_line(line_name="root-pos-x", line_data=data.qpos[0], fig_idx=0)
-    viewer.add_data_to_line(line_name="root-pos-z", line_data=data.qpos[2], fig_idx=0)
+    viewer.add_data_to_line(line_name="root-pos-x",
+                            line_data=data.qpos[0], fig_idx=0)
+    viewer.add_data_to_line(line_name="root-pos-z",
+                            line_data=data.qpos[2], fig_idx=0)
     viewer.add_data_to_line(line_name="right_ankle_y", line_data=data.qpos[
         mujoco.mj_name2id(model, mujoco.mjtObj.mjOBJ_JOINT, "right_ankle_y")], fig_idx=1)
 
@@ -41,4 +43,3 @@
 
 # close
 viewer.close()
-

mujoco_viewer/mujoco_viewer.py

@@ -132,7 +132,7 @@ def __init__(
         self._overlay = {}
         self._markers = []
 
-    def add_line_to_fig(self, line_name, fig_idx = 0):
+    def add_line_to_fig(self, line_name, fig_idx=0):
         assert isinstance(line_name, str), \
             "Line name must be a string."
 
@@ -156,7 +156,7 @@ def add_line_to_fig(self, line_name, fig_idx = 0):
         for i in range(mujoco.mjMAXLINEPNT):
             fig.linedata[linecount][2*i] = -float(i)
 
-    def add_data_to_line(self, line_name, line_data, fig_idx = 0):
+    def add_data_to_line(self, line_name, line_data, fig_idx=0):
         fig = self.figs[fig_idx]
 
         try:
@@ -174,8 +174,8 @@ def add_data_to_line(self, line_name, line_data, fig_idx = 0):
             fig.linedata[line_idx][2*i + 1] = fig.linedata[line_idx][2*i - 1]
 
         # assign new
-        fig.linepnt[line_idx] = pnt;
-        fig.linedata[line_idx][1] = line_data;
+        fig.linepnt[line_idx] = pnt
+        fig.linedata[line_idx][1] = line_data
 
     def add_marker(self, **marker_params):
         self._markers.append(marker_params)
@@ -357,7 +357,7 @@ def read_pixels(self, camid=None, depth=False):
         # render
         mujoco.mjr_render(self.viewport, self.scn, self.ctx)
         shape = glfw.get_framebuffer_size(self.window)
-        
+
         if depth:
             rgb_img = np.zeros((shape[1], shape[0], 3), dtype=np.uint8)
             depth_img = np.zeros((shape[1], shape[0], 1), dtype=np.float32)
@@ -368,7 +368,6 @@ def read_pixels(self, camid=None, depth=False):
             mujoco.mjr_readPixels(img, None, self.viewport, self.ctx)
             return np.flipud(img)
 
-
     def render(self):
         if self.render_mode == 'offscreen':
             raise NotImplementedError(
@@ -429,7 +428,7 @@ def update():
                         viewport = mujoco.MjrRect(
                             x, y, int(width / 4), int(height / 4))
 
-                        has_lines = len([i for i in fig.linename if i!=b''])
+                        has_lines = len([i for i in fig.linename if i != b''])
                         if has_lines:
                             mujoco.mjr_figure(viewport, fig, self.ctx)