format

.style.yapf

@@ -0,0 +1,7 @@
+[style]
+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

format.sh

@@ -0,0 +1,20 @@
+#!/usr/bin/env bash
+# Usage: at the root dir >> bash scripts/format.sh
+
+# Check yapf version.
+ver=$(yapf --version)
+if ! echo $ver | grep -q 0.30.0; then
+  echo "Wrong YAPF version installed: 0.30.0 is required, not $ver. Please install via `pip install yapf==0.30.0`"
+  exit 1
+fi
+
+yapf --in-place --recursive -p --verbose --style .style.yapf scenarionet/
+
+if [[ "$1" == '--test' ]]; then # Only for CI usage, user should not use --test flag.
+  if ! git diff --quiet &>/dev/null; then
+    echo '*** You have not formatted your code! Please run [bash format.sh] at root directory before commit! Thanks! ***'
+    exit 1
+  else
+    echo "Code style test passed!"
+  fi
+fi

scenarionet/builder/select_demo_pickles.py

@@ -1,6 +1,5 @@
 import pickle
 
-
 if __name__ == '__main__':
 
     with open("waymo120/0408_output_final/dataset_summary.pkl", "rb") as f:

scenarionet/converter/utils.py

@@ -66,13 +66,9 @@ def contains_explicit_return(f):
     return any(isinstance(node, ast.Return) for node in ast.walk(ast.parse(inspect.getsource(f))))
 
 
-def write_to_directory(convert_func,
-                       scenarios,
-                       output_path,
-                       dataset_version,
-                       dataset_name,
-                       force_overwrite=False,
-                       **kwargs):
+def write_to_directory(
+    convert_func, scenarios, output_path, dataset_version, dataset_name, force_overwrite=False, **kwargs
+):
     """
     Convert a batch of scenarios.
     """

scenarionet/examples/convert_nuplan.py

@@ -19,7 +19,8 @@ if __name__ == "__main__":
     # scenarios = get_nuplan_scenarios(data_root, map_root, logs=["2021.07.16.20.45.29_veh-35_01095_01486"])
     scenarios = get_nuplan_scenarios(data_root, map_root)
 
-    write_to_directory(convert_func=convert_nuplan_scenario,
+    write_to_directory(
+        convert_func=convert_nuplan_scenario,
         scenarios=scenarios,
         output_path=output_path,
         dataset_version=version,

scenarionet/examples/convert_nuscenes.py

@@ -17,10 +17,12 @@ if __name__ == "__main__":
     dataroot = '/home/shady/data/nuscenes'
     scenarios, nusc = get_nuscenes_scenarios(dataroot, version)
 
-    write_to_directory(convert_func=convert_nuscenes_scenario,
+    write_to_directory(
+        convert_func=convert_nuscenes_scenario,
         scenarios=scenarios,
         output_path=output_path,
         dataset_version=version,
         dataset_name=dataset_name,
         force_overwrite=force_overwrite,
-                       nuscenes=nusc)
+        nuscenes=nusc
+    )

scenarionet/examples/convert_waymo.py

@@ -17,9 +17,11 @@ if __name__ == '__main__':
     waymo_data_direction = os.path.join(SCENARIONET_DATASET_PATH, "waymo_origin")
     scenarios = get_waymo_scenarios(waymo_data_direction)
 
-    write_to_directory(convert_func=convert_waymo_scenario,
+    write_to_directory(
+        convert_func=convert_waymo_scenario,
         scenarios=scenarios,
         output_path=output_path,
         dataset_version=version,
         dataset_name=dataset_name,
-                       force_overwrite=force_overwrite)
+        force_overwrite=force_overwrite
+    )

setup.py

@@ -36,6 +36,7 @@ install_requires = [
     "tqdm",
     "metadrive-simulator",
     "geopandas",
+    "yapf==0.30.0",
     "shapely"
 ]