Merge pull request #8 from opendilab/gl-dev

Fix code format

Author: RobinC94

.gitignore

@@ -1405,3 +1405,6 @@ htmlcov
 project_test.py
 *.csv
 *.avi
+*episode_metainfo.json
+*measurements.lmdb
+*index.txt

.style.yapf

@@ -0,0 +1,11 @@
+[style]
+# For explanation and more information: https://github.com/google/yapf
+BASED_ON_STYLE=pep8
+DEDENT_CLOSING_BRACKETS=True
+SPLIT_BEFORE_FIRST_ARGUMENT=True
+ALLOW_SPLIT_BEFORE_DICT_VALUE=False
+JOIN_MULTIPLE_LINES=False
+COLUMN_LIMIT=120
+BLANK_LINE_BEFORE_NESTED_CLASS_OR_DEF=True
+BLANK_LINES_AROUND_TOP_LEVEL_DEFINITION=2
+SPACES_AROUND_POWER_OPERATOR=True

core/data/dataset_saver.py

@@ -10,7 +10,6 @@
 from core.utils.others.image_helper import read_image, draw_msra_gaussian
 from core.utils.data_utils import augmenter
 
-
 PIXEL_OFFSET = 10
 PIXELS_PER_METER = 5
 
@@ -160,12 +159,7 @@ def __getitem__(self, index) -> Any:
             regression_offset[i] = center - center_int
             indices[i] = center_int[1] * output_size + center_int[0]
 
-        return {
-            'birdview': birdview,
-            'location': np.array(locations),
-            'command': command,
-            'speed': speed
-        }
+        return {'birdview': birdview, 'location': np.array(locations), 'command': command, 'speed': speed}
 
 
 class LBCImageDataset(Dataset):
@@ -196,7 +190,7 @@ def __init__(
         self._batch_read_number = batch_read_number
         self._batch_aug = batch_aug
 
-        print ("augment with ", augment_strategy)
+        print("augment with ", augment_strategy)
         if augment_strategy is not None and augment_strategy != 'None':
             self.augmenter = getattr(augmenter, augment_strategy)
         else:
@@ -240,7 +234,9 @@ def __getitem__(self, index) -> Any:
         rgb_image = read_image(png_file).reshape(160, 384, 3)
 
         if self.augmenter:
-            rgb_images = [self.augmenter(self._batch_read_number).augment_image(rgb_image) for i in range(self._batch_aug)]
+            rgb_images = [
+                self.augmenter(self._batch_read_number).augment_image(rgb_image) for i in range(self._batch_aug)
+            ]
         else:
             rgb_images = [rgb_image for i in range(self._batch_aug)]
 

core/data/lbc_dataset.py

@@ -36,7 +36,7 @@ class LBCBirdviewPolicy(BaseCarlaPolicy):
     config = dict(
         cuda=True,
         model=dict(),
-        learn=dict(loss='l1',),
+        learn=dict(loss='l1', ),
         steer_points=None,
         pid=None,
         gap=5,

core/policy/lbc_policy.py

@@ -0,0 +1,3 @@
+_logs
+_preloads
+vis