1 files changed, 33 insertions, 32 deletions
diff --git a/examples/pybullet/gym/pybullet_envs/deep_mimic/mocap/mocap_dataset.py b/examples/pybullet/gym/pybullet_envs/deep_mimic/mocap/mocap_dataset.py
index 18bc27ae4..c297220db 100644
--- a/examples/pybullet/gym/pybullet_envs/deep_mimic/mocap/mocap_dataset.py
+++ b/examples/pybullet/gym/pybullet_envs/deep_mimic/mocap/mocap_dataset.py
@@ -1,36 +1,37 @@
 import numpy as np
 from skeleton import Skeleton
 
+
 class MocapDataset:
-    def __init__(self, fps, skeleton):
-        self._skeleton = skeleton
-        self._fps = fps
-        self._data = None # Must be filled by subclass
-        self._cameras = None # Must be filled by subclass
-    
-    def remove_joints(self, joints_to_remove):
-        kept_joints = self._skeleton.remove_joints(joints_to_remove)
-        for subject in self._data.keys():
-            for action in self._data[subject].keys():
-                s = self._data[subject][action]
-                s['positions'] = s['positions'][:, kept_joints]
-                
-        
-    def __getitem__(self, key):
-        return self._data[key]
-        
-    def subjects(self):
-        return self._data.keys()
-    
-    def fps(self):
-        return self._fps
-    
-    def skeleton(self):
-        return self._skeleton
-        
-    def cameras(self):
-        return self._cameras
-    
-    def supports_semi_supervised(self):
-        # This method can be overridden
-        return False 
+
+  def __init__(self, fps, skeleton):
+    self._skeleton = skeleton
+    self._fps = fps
+    self._data = None  # Must be filled by subclass
+    self._cameras = None  # Must be filled by subclass
+
+  def remove_joints(self, joints_to_remove):
+    kept_joints = self._skeleton.remove_joints(joints_to_remove)
+    for subject in self._data.keys():
+      for action in self._data[subject].keys():
+        s = self._data[subject][action]
+        s['positions'] = s['positions'][:, kept_joints]
+
+  def __getitem__(self, key):
+    return self._data[key]
+
+  def subjects(self):
+    return self._data.keys()
+
+  def fps(self):
+    return self._fps
+
+  def skeleton(self):
+    return self._skeleton
+
+  def cameras(self):
+    return self._cameras
+
+  def supports_semi_supervised(self):
+    # This method can be overridden
+    return False